Merge pull request #391 from BaiMoHan/main

fix: ndarray serialized & ignore save cn_image
mrhan1993 · Aug 6, 2024 · e3ad26a · e3ad26a
2 parents b2aa7ac + 4bd21c6
commit e3ad26a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/fooocusapi/sql_client.py b/fooocusapi/sql_client.py
@@ -151,6 +151,11 @@ def store_history(self, record: dict) -> None:
         :param record:
         :return:
         """
+        serialized_image_prompts = [
+            (cn_stop, cn_wight, cn_type)
+            for arr, cn_stop, cn_wight, cn_type in record['image_prompts']
+        ]
+        record['image_prompts'] = serialized_image_prompts
         self.session.add_all([GenerateRecord(**record)])
         self.session.commit()