yunusajib commited on
Commit
dd2c70a
·
verified ·
1 Parent(s): 59ed260

update llava model

Browse files
Files changed (1) hide show
  1. llava_inference.py +1 -1
llava_inference.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
 
6
  class LLaVAHelper:
7
  def __init__(self, model_name="llava-hf/llava-1.5-7b-hf"):
8
- self.tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  self.model, self.image_processor, _ = load_pretrained_model(model_name, None)
10
  self.model.eval()
11
 
 
5
 
6
  class LLaVAHelper:
7
  def __init__(self, model_name="llava-hf/llava-1.5-7b-hf"):
8
+ self.tokenizer = AutoTokenizer.from_pretrained(model_name, force_download=True)
9
  self.model, self.image_processor, _ = load_pretrained_model(model_name, None)
10
  self.model.eval()
11