Spaces:
Runtime error
Runtime error
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)
|