Spaces:
Running
Running
jiang
commited on
Commit
•
369592b
1
Parent(s):
324558f
update
Browse files
demo.py
CHANGED
@@ -21,7 +21,7 @@ use_fp16 = True
|
|
21 |
# Load pretrained ckpt & config
|
22 |
overrides={"bpe_dir":"utils/BPE"}
|
23 |
models, cfg, task = load_model_ensemble_and_task(
|
24 |
-
utils.split_paths('
|
25 |
arg_overrides=overrides
|
26 |
)
|
27 |
# print(cfg)
|
|
|
21 |
# Load pretrained ckpt & config
|
22 |
overrides={"bpe_dir":"utils/BPE"}
|
23 |
models, cfg, task = load_model_ensemble_and_task(
|
24 |
+
utils.split_paths('polyformer_l_refcocog.pt'),
|
25 |
arg_overrides=overrides
|
26 |
)
|
27 |
# print(cfg)
|