RoversX commited on
Commit
21217e9
1 Parent(s): 27d8931

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("cuda")
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()