Spaces:
Runtime error
Runtime error
Commit
•
359e303
1
Parent(s):
57d1c86
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ from PIL import Image
|
|
18 |
|
19 |
|
20 |
tokenizer = AutoTokenizer.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', use_fast=False)
|
21 |
-
model = LlavaLlamaForCausalLM.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', low_cpu_mem_usage=True,
|
22 |
image_processor = model.get_vision_tower()
|
23 |
if not vision_tower.is_loaded:
|
24 |
vision_tower.load_model()
|
|
|
18 |
|
19 |
|
20 |
tokenizer = AutoTokenizer.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', use_fast=False)
|
21 |
+
model = LlavaLlamaForCausalLM.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B', low_cpu_mem_usage=True,torch_dtype=torch.float16)
|
22 |
image_processor = model.get_vision_tower()
|
23 |
if not vision_tower.is_loaded:
|
24 |
vision_tower.load_model()
|