Update prepare.py

VinAIResearch · Jun 27, 2024 · 15509e8 · 15509e8
1 parent c932274
commit 15509e8
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/prepare.py b/prepare.py
@@ -12,10 +12,7 @@
 
 
 def process_prompt_data(index, batch_start, prompt_embed, output_path):
-    prompt_dict = {
-        "prompt_embeds": prompt_embed,
-    }
-    np.save(output_path / f"{batch_start+index}.npy", **prompt_dict)
+    np.save(output_path / f"{batch_start+index}.npy", prompt_embed)
     return index