From c9322748e3f457e1ad7fabdf7c30b83fd2c49677 Mon Sep 17 00:00:00 2001
From: "Thuan H. Nguyen" <32274287+thuanz123@users.noreply.github.com>
Date: Thu, 27 Jun 2024 11:12:37 +0700
Subject: [PATCH] Update prepare.py

---
 prepare.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prepare.py b/prepare.py
index 9236549..0635236 100644
--- 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