From 62f6da3098e71d9f5e229dccd76a6ffc162270cd Mon Sep 17 00:00:00 2001 From: "Zhe Zhao (George)" Date: Tue, 23 Jan 2024 11:57:59 +0800 Subject: [PATCH] Update demo.py to fix the override of llama_type --- llama_adapter_v2_multimodal7b/demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_adapter_v2_multimodal7b/demo.py b/llama_adapter_v2_multimodal7b/demo.py index d7877a8..78e3a97 100644 --- a/llama_adapter_v2_multimodal7b/demo.py +++ b/llama_adapter_v2_multimodal7b/demo.py @@ -8,7 +8,7 @@ llama_dir = "/path/to/LLaMA/" # choose from BIAS-7B, LORA-BIAS-7B, CAPTION-7B.pth -model, preprocess = llama.load("BIAS-7B", llama_dir, device) +model, preprocess = llama.load("BIAS-7B", llama_dir, device=device) model.eval() prompt = llama.format_prompt('Please introduce this painting.')