Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|