haowu11 commited on
Commit
0ff530e
·
verified ·
1 Parent(s): d37c06b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -95,6 +95,7 @@ pipe.enable_model_cpu_offload()
95
 
96
  @spaces.GPU
97
  def infer(face_img,pose_img, garm_img, prompt,negative_prompt, n_samples, n_steps, seed):
 
98
  face_img = Image.open(face_img)
99
  pose_img = Image.open(pose_img)
100
  garm_img = Image.open(garm_img)
 
95
 
96
  @spaces.GPU
97
  def infer(face_img,pose_img, garm_img, prompt,negative_prompt, n_samples, n_steps, seed):
98
+ pipe.face_clip_encoder.to('cuda')
99
  face_img = Image.open(face_img)
100
  pose_img = Image.open(pose_img)
101
  garm_img = Image.open(garm_img)