Spaces:

rlawjdghek
/

StableVITON

Running on Zero

rlawjdghek commited on Apr 1

Commit

107d5d9

•

1 Parent(s): d361875

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-import spaces
 import os
 import sys
 import time
@@ -87,7 +87,7 @@ def stable_viton_model_hd(
  pil_output = Image.fromarray(output)
  return pil_output
-@spaces.GPU # TODO: turn on when final upload
 @torch.no_grad()
 def process_hd(vton_img, garm_img, n_steps):
  model_type = 'hd'

+# import spaces
 import os
 import sys
 import time
  pil_output = Image.fromarray(output)
  return pil_output
+# @spaces.GPU # TODO: turn on when final upload
 @torch.no_grad()
 def process_hd(vton_img, garm_img, n_steps):
  model_type = 'hd'