Safetensors
llava
k-m-irfan commited on
Commit
be250de
·
verified ·
1 Parent(s): 614053a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
  )