Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
|
| 31 |
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
| 32 |
torch_dtype=torch.float16,
|
| 33 |
)
|
| 34 |
-
refiner.to("
|
| 35 |
|
| 36 |
|
| 37 |
model = create_model("./models/cldm_v15_unicontrol.yaml").cpu()
|
|
|
|
| 31 |
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
| 32 |
torch_dtype=torch.float16,
|
| 33 |
)
|
| 34 |
+
refiner.to("cuda")
|
| 35 |
|
| 36 |
|
| 37 |
model = create_model("./models/cldm_v15_unicontrol.yaml").cpu()
|