lalalalalalalalalala commited on
Commit
12c9224
·
verified ·
1 Parent(s): 3a4995a

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -23,7 +23,7 @@ def load_hf_dataset(dataset_path, auth_token):
23
  video_paths = dataset
24
  return video_paths
25
 
26
- def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, endpoint, video_src, video_hf, video_hf_auth, parquet_index, video_od, video_od_auth, video_gd, video_gd_auth, frame_format, frame_limit):
27
  progress_info = []
28
  with tempfile.TemporaryDirectory() as temp_dir:
29
  # temp_dir = '/opt/run'
 
23
  video_paths = dataset
24
  return video_paths
25
 
26
+ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, endpoint, video_hf, video_hf_auth, parquet_index, video_od, video_od_auth, video_gd, video_gd_auth, frame_format, frame_limit):
27
  progress_info = []
28
  with tempfile.TemporaryDirectory() as temp_dir:
29
  # temp_dir = '/opt/run'