jiang commited on
Commit
369592b
1 Parent(s): 324558f
Files changed (1) hide show
  1. demo.py +1 -1
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('weights/polyformer_l_refcocog.pt'),
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)