Spaces:

RoversX
/

sketch-to-image

Runtime error

RoversX commited on Sep 1, 2023

Commit

9c00e77

1 Parent(s): 21217e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-refiner-1.0",
     torch_dtype=torch.float16,
 )
-refiner.to("cpu")
 model = create_model("./models/cldm_v15_unicontrol.yaml").cpu()

     "stabilityai/stable-diffusion-xl-refiner-1.0",
     torch_dtype=torch.float16,
 )
+refiner.to("cuda")
 model = create_model("./models/cldm_v15_unicontrol.yaml").cpu()