Update run.py

run.py

@@ -27,7 +27,8 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
     progress_info = []
     with tempfile.TemporaryDirectory() as temp_dir:
         # temp_dir = '/opt/run'
-        csv_filename = os.path.join(temp_dir, 'caption.csv')
+        # csv_filename = os.path.join(temp_dir, 'caption.csv')
+        csv_filename = '/dev/shm/caption.csv'
         print(csv_filename)
         with open(csv_filename, mode='w', newline='') as csv_file:
             fieldnames = ['md5', 'caption']
@@ -42,7 +43,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                     repo_id=video_hf,
                     filename='data/' + str(parquet_index).zfill(6) + '.parquet',
                     repo_type="dataset",
-                    #local_dir="/dev/shm",
+                    local_dir="/dev/shm",
                     token=video_hf_auth,
                 )