ResearcherXman commited on
Commit
da665db
1 Parent(s): 3ddd9a0
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -131,6 +131,9 @@ pipe.scheduler = diffusers.EulerDiscreteScheduler.from_config(
131
 
132
  pipe.cuda()
133
  pipe.load_ip_adapter_instantid(face_adapter)
 
 
 
134
  # load and disable LCM
135
  pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
136
  pipe.disable_lora()
 
131
 
132
  pipe.cuda()
133
  pipe.load_ip_adapter_instantid(face_adapter)
134
+ print(pipe.image_proj_model.device, pipe.unet.device)
135
+ pipe.image_proj_model.to("cuda")
136
+ pipe.unet.to("cuda")
137
  # load and disable LCM
138
  pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
139
  pipe.disable_lora()