lalalalalalalalalala commited on
Commit
0228656
β€’
1 Parent(s): bca51b0

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -0
run.py CHANGED
@@ -20,6 +20,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
20
  progress_info = []
21
  with tempfile.TemporaryDirectory() as temp_dir:
22
  csv_filename = os.path.join(temp_dir, str(parquet_index) + '_caption.csv')
 
23
  with open(csv_filename, mode='w', newline='') as csv_file:
24
  fieldnames = ['md5', 'caption']
25
  writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
 
20
  progress_info = []
21
  with tempfile.TemporaryDirectory() as temp_dir:
22
  csv_filename = os.path.join(temp_dir, str(parquet_index) + '_caption.csv')
23
+ print(csv_filename)
24
  with open(csv_filename, mode='w', newline='') as csv_file:
25
  fieldnames = ['md5', 'caption']
26
  writer = csv.DictWriter(csv_file, fieldnames=fieldnames)