rlawjdghek commited on
Commit
bd2a249
β€’
1 Parent(s): 4e09d36
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- # import spaces
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
- # @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'
 
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'