fix gc collect

Fcucgvhhhvjv · Oct 7, 2022 · 7c0f602 · 7c0f602
1 parent 945107e
commit 7c0f602
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/stable-diffusion-streamlit/pages/文字转图片.py b/src/stable-diffusion-streamlit/pages/文字转图片.py
@@ -5,6 +5,7 @@
 from datetime import datetime
 import json
 import os
+import gc
 from diffusers.training_utils import set_seed
 
 from pages.model.inference import result_dir, quant_pipe
@@ -129,6 +130,8 @@
         json_filename = os.path.join(result_dir, "text2image", uid, "config.json")
         os.makedirs(os.path.join(result_dir, "text2image", uid), exist_ok=True)
         t.get_result().images[0].save(image_filename)
+        del t
+        gc.collect()
         with open(json_filename, "w") as f:
             config = {
                 "text_prompt": text_prompt,