Update README.md
Browse files
README.md
CHANGED
@@ -67,7 +67,7 @@ def infer(
|
|
67 |
load_8bit=False,
|
68 |
device="cuda"
|
69 |
):
|
70 |
-
model_name = get_model_name_from_path(model_path)+"llava_qwen"
|
71 |
tokenizer, model, image_processor, context_len = load_pretrained_model(
|
72 |
model_path, model_base, model_name, load_8bit=load_8bit
|
73 |
)
|
|
|
67 |
load_8bit=False,
|
68 |
device="cuda"
|
69 |
):
|
70 |
+
model_name = get_model_name_from_path(model_path)+"llava_qwen" # For llava internal checks and proper loading
|
71 |
tokenizer, model, image_processor, context_len = load_pretrained_model(
|
72 |
model_path, model_base, model_name, load_8bit=load_8bit
|
73 |
)
|