Spaces:
Build error
Build error
rlawjdghek
commited on
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'
|