Spaces:

jjz5463
/

Diary-AI

Paused

jjz5463 commited on Sep 29, 2024

Commit

fd53ea7

1 Parent(s): 8127d0e

GUI debug

Files changed (2) hide show

Experiments/Baseline/baseline_utils.py CHANGED Viewed

@@ -77,7 +77,7 @@ def analyze_writer_image(image_path, api_key):
 def generate_comic_book(diary_text, writer_description, num_pages=4):
     pipe = AutoPipelineForText2Image.from_pretrained(
         "stabilityai/sdxl-turbo",
-        #torch_dtype=torch.float16,
         variant="fp16",
         cache_dir="./SDXL-Turbo"
     )

 def generate_comic_book(diary_text, writer_description, num_pages=4):
     pipe = AutoPipelineForText2Image.from_pretrained(
         "stabilityai/sdxl-turbo",
+        torch_dtype=torch.float16,
         variant="fp16",
         cache_dir="./SDXL-Turbo"
     )

Experiments/Baseline/{GUI.py → streamlit_GUI.py} RENAMED Viewed

File without changes