haowu11 commited on
Commit
0f2c0d7
·
verified ·
1 Parent(s): 92d69f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -88,6 +88,7 @@ pipe.set_face_fidelity_scale(ip_scale)
88
  pipe = pipe.to("cuda")
89
  pipe.enable_model_cpu_offload()
90
 
 
91
  def infer(face_img,pose_img, garm_img, prompt,negative_prompt, n_samples, n_steps, seed):
92
  face_img = Image.open(face_img)
93
  pose_img = Image.open(pose_img)
 
88
  pipe = pipe.to("cuda")
89
  pipe.enable_model_cpu_offload()
90
 
91
+ @sapces.GPU
92
  def infer(face_img,pose_img, garm_img, prompt,negative_prompt, n_samples, n_steps, seed):
93
  face_img = Image.open(face_img)
94
  pose_img = Image.open(pose_img)