Spaces:
Build error
Build error
Commit
Β·
bd2a249
1
Parent(s):
4e09d36
add zero
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
import os
|
| 3 |
import sys
|
| 4 |
import time
|
|
@@ -94,7 +94,7 @@ def stable_viton_model_hd(
|
|
| 94 |
pil_output = Image.fromarray(output)
|
| 95 |
return pil_output
|
| 96 |
|
| 97 |
-
|
| 98 |
@torch.no_grad()
|
| 99 |
def process_hd(vton_img, garm_img, n_steps):
|
| 100 |
model_type = 'hd'
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import os
|
| 3 |
import sys
|
| 4 |
import time
|
|
|
|
| 94 |
pil_output = Image.fromarray(output)
|
| 95 |
return pil_output
|
| 96 |
|
| 97 |
+
@spaces.GPU # TODO: turn on when final upload
|
| 98 |
@torch.no_grad()
|
| 99 |
def process_hd(vton_img, garm_img, n_steps):
|
| 100 |
model_type = 'hd'
|