From aaff4d9ebbfa441059696787979543ba7c38b3ed Mon Sep 17 00:00:00 2001
From: 007gzs <007gzs@gmail.com>
Date: Thu, 19 Sep 2024 09:44:08 +0800
Subject: [PATCH] =?UTF-8?q?fix=20prompt=5Ftext=20=E4=B8=AD=E6=9C=89<>?=
 =?UTF-8?q?=E6=97=B6=E4=BF=9D=E5=AD=98=E5=A4=B1=E8=B4=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 yijian_community/data/data_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/yijian_community/data/data_utils.py b/yijian_community/data/data_utils.py
index 384027e..29f0a96 100644
--- a/yijian_community/data/data_utils.py
+++ b/yijian_community/data/data_utils.py
@@ -16,6 +16,7 @@
 
 import os
 from typing import List
+import hashlib
 
 import pandas as pd
 from datasets import Dataset, load_dataset
@@ -88,7 +89,7 @@ def save_image(
     """
     img_save_paths = []
     for prompt_text, image in zip(prompt_texts, images):
-        img_save_path = os.path.join(save_path, prompt_text[:50] + ".jpg")
+        img_save_path = os.path.join(save_path, hashlib.md5(prompt_text.encode("utf8")).hexdigest() + ".jpg")
         image.save(img_save_path)
         img_save_paths.append(img_save_path)
     return img_save_paths