wufan commited on
Commit
acd09f6
1 Parent(s): 906b977

Update cfg_small.yaml

Browse files
Files changed (1) hide show
  1. cfg_small.yaml +3 -3
cfg_small.yaml CHANGED
@@ -2,13 +2,13 @@ model:
2
  arch: unimernet
3
  model_type: unimernet
4
  model_config:
5
- model_name: MODEL_DIR
6
  max_seq_len: 1536
7
 
8
  load_pretrained: True
9
- pretrained: 'MODEL_DIR/unimernet_small.pth'
10
  tokenizer_config:
11
- path: MODEL_DIR
12
 
13
  datasets:
14
  formula_rec_eval:
 
2
  arch: unimernet
3
  model_type: unimernet
4
  model_config:
5
+ model_name: ./models/unimernet_small
6
  max_seq_len: 1536
7
 
8
  load_pretrained: True
9
+ pretrained: './models/unimernet_small/unimernet_small.pth'
10
  tokenizer_config:
11
+ path: ./models/unimernet_small
12
 
13
  datasets:
14
  formula_rec_eval: