Update prepare.py

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