From a8572190a85cbd651259665c8bbbe9145bd583aa Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Sun, 27 Apr 2025 18:38:51 +0800 Subject: [PATCH] update eval --- eval_vlm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eval_vlm.py b/eval_vlm.py index 631fbb0..850eeff 100644 --- 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'