Spaces:
Running
on
Zero
Running
on
Zero
kxhit
commited on
Commit
·
319afdb
1
Parent(s):
0e0b8fc
run successfully & move carvkit to cuda
Browse files
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=
|
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)
|