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
|