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("cpu")
|
35 |
|
36 |
|
37 |
model = create_model("./models/cldm_v15_unicontrol.yaml").cpu()
|