Spaces:
Sleeping
Sleeping
update llava model
Browse files- 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 |
|