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)
|