GUI debug
Browse files
Experiments/Baseline/baseline_utils.py
CHANGED
|
@@ -77,7 +77,7 @@ def analyze_writer_image(image_path, api_key):
|
|
| 77 |
def generate_comic_book(diary_text, writer_description, num_pages=4):
|
| 78 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
| 79 |
"stabilityai/sdxl-turbo",
|
| 80 |
-
|
| 81 |
variant="fp16",
|
| 82 |
cache_dir="./SDXL-Turbo"
|
| 83 |
)
|
|
|
|
| 77 |
def generate_comic_book(diary_text, writer_description, num_pages=4):
|
| 78 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
| 79 |
"stabilityai/sdxl-turbo",
|
| 80 |
+
torch_dtype=torch.float16,
|
| 81 |
variant="fp16",
|
| 82 |
cache_dir="./SDXL-Turbo"
|
| 83 |
)
|
Experiments/Baseline/{GUI.py → streamlit_GUI.py}
RENAMED
|
File without changes
|