lalalalalalalalalala commited on
Commit
8b7cc82
·
verified ·
1 Parent(s): d1adc1d

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -38,7 +38,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
38
  with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_parquet_file:
39
  temp_parquet_file = hf_hub_download(
40
  repo_id=video_hf,
41
- filename='data/' + str(number).zfill(6) + '.parquet',
42
  repo_type="dataset",
43
  token=video_hf_auth,
44
  )
 
38
  with tempfile.NamedTemporaryFile(mode='w+t', delete=True) as temp_parquet_file:
39
  temp_parquet_file = hf_hub_download(
40
  repo_id=video_hf,
41
+ filename='data/' + str(parquet_index).zfill(6) + '.parquet',
42
  repo_type="dataset",
43
  token=video_hf_auth,
44
  )