fixing bugs
Browse files
app.py
CHANGED
@@ -85,8 +85,7 @@ try:
|
|
85 |
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
86 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
87 |
torch_dtype=torch_dtype,
|
88 |
-
|
89 |
-
variant="fp16" if device == "cuda" else None
|
90 |
).to(device)
|
91 |
|
92 |
# LoRA
|
|
|
85 |
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
86 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
87 |
torch_dtype=torch_dtype,
|
88 |
+
variant="fp32"
|
|
|
89 |
).to(device)
|
90 |
|
91 |
# LoRA
|