Spaces:

koajoel
/

PolyFormer

Running

jiang commited on May 16, 2023

Commit

369592b

1 Parent(s): 324558f

update

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -21,7 +21,7 @@ use_fp16 = True
 # Load pretrained ckpt & config
 overrides={"bpe_dir":"utils/BPE"}
 models, cfg, task = load_model_ensemble_and_task(
-        utils.split_paths('weights/polyformer_l_refcocog.pt'),
         arg_overrides=overrides
     )
 # print(cfg)

 # Load pretrained ckpt & config
 overrides={"bpe_dir":"utils/BPE"}
 models, cfg, task = load_model_ensemble_and_task(
+        utils.split_paths('polyformer_l_refcocog.pt'),
         arg_overrides=overrides
     )
 # print(cfg)