Spaces:

opendatalab
/

UniMERNet-Demo

Running on Zero

wufan commited on Sep 18

Commit

6fa1649

•

1 Parent(s): e907bd9

Update cfg_base.yaml

Files changed (1) hide show

cfg_base.yaml CHANGED Viewed

@@ -6,7 +6,7 @@ model:
     max_seq_len: 1536
   load_pretrained: True
-  pretrained: './models/unimernet_base/unimernet_base.pth'
   tokenizer_config:
     path: ./models/unimernet_base

     max_seq_len: 1536
   load_pretrained: True
+  pretrained: './models/unimernet_base/pytorch_model.pth'
   tokenizer_config:
     path: ./models/unimernet_base