kxhit commited on
Commit
319afdb
·
1 Parent(s): 0e0b8fc

run successfully & move carvkit to cuda

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -136,7 +136,7 @@ def create_carvekit_interface():
136
  interface = HiInterface(object_type="object", # Can be "object" or "hairs-like".
137
  batch_size_seg=6,
138
  batch_size_matting=1,
139
- device="cpu",
140
  seg_mask_size=640, # Use 640 for Tracer B7 and 320 for U2Net
141
  matting_mask_size=2048,
142
  trimap_prob_threshold=231,
@@ -153,7 +153,7 @@ predictor = sam_init()
153
 
154
 
155
 
156
- @spaces.GPU
157
  def run_eschernet(eschernet_input_dict, sample_steps, sample_seed, nvs_num, nvs_mode):
158
  # set the random seed
159
  generator = torch.Generator(device=device).manual_seed(sample_seed)
 
136
  interface = HiInterface(object_type="object", # Can be "object" or "hairs-like".
137
  batch_size_seg=6,
138
  batch_size_matting=1,
139
+ device=device,
140
  seg_mask_size=640, # Use 640 for Tracer B7 and 320 for U2Net
141
  matting_mask_size=2048,
142
  trimap_prob_threshold=231,
 
153
 
154
 
155
 
156
+ @spaces.GPU(duration=120)
157
  def run_eschernet(eschernet_input_dict, sample_steps, sample_seed, nvs_num, nvs_mode):
158
  # set the random seed
159
  generator = torch.Generator(device=device).manual_seed(sample_seed)