From e5f222c05720b2101144deeba9376c7b241c6cb1 Mon Sep 17 00:00:00 2001 From: Ryan Peruski <91974124+Silverasdf@users.noreply.github.com> Date: Mon, 25 Nov 2024 16:05:45 -0500 Subject: [PATCH] Fixed argument order so that load_pretrained_model() works Argument order is different in LLaVA's function, so I updated it so that it doesn't matter which order the arguments are in. --- llava/eval/run_vila.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llava/eval/run_vila.py b/llava/eval/run_vila.py index dadac279..ca7e2c1f 100755 --- a/llava/eval/run_vila.py +++ b/llava/eval/run_vila.py @@ -65,7 +65,7 @@ def eval_model(args): images, num_frames = opencv_extract_frames(video_file, args.num_video_frames) model_name = get_model_name_from_path(args.model_path) - tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, model_name, args.model_base) + tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, model_name=model_name, model_base=args.model_base) qs = args.query image_token_se = DEFAULT_IM_START_TOKEN + DEFAULT_IMAGE_TOKEN + DEFAULT_IM_END_TOKEN