update eval

2025-04-27 18:38:51 +08:00 · 2025-04-27 18:38:51 +08:00 · a8572190a8
parent 623a130c34
commit a8572190a8
1 changed files with 1 additions and 1 deletions
--- a/eval_vlm.py
+++ b/eval_vlm.py
@ -16,8 +16,8 @@ def count_parameters(model):


 def init_model(lm_config, device):
-    tokenizer = AutoTokenizer.from_pretrained('./model')
    if args.load == 0:
+        tokenizer = AutoTokenizer.from_pretrained('./model')
        moe_path = '_moe' if args.use_moe else ''
        modes = {0: 'pretrain_vlm', 1: 'sft_vlm', 2: 'sft_vlm_multi'}
        ckp = f'./{args.out_dir}/{modes[args.model_mode]}_{args.hidden_size}{moe_path}.pth'