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 |
)
|