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