change back to float 16
Browse files- baseline_utils.py +2 -2
baseline_utils.py
CHANGED
@@ -77,8 +77,8 @@ 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 |
-
torch_dtype=torch.
|
81 |
-
variant="
|
82 |
cache_dir="./SDXL-Turbo"
|
83 |
)
|
84 |
|
|
|
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 |
)
|
84 |
|