Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,8 @@ from PIL import Image
|
|
| 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 |
-
|
|
|
|
| 23 |
if not vision_tower.is_loaded:
|
| 24 |
vision_tower.load_model()
|
| 25 |
vision_tower.to(device=device, dtype=torch.float16)
|
|
|
|
| 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 |
+
|
| 23 |
+
vision_tower = model.get_vision_tower()
|
| 24 |
if not vision_tower.is_loaded:
|
| 25 |
vision_tower.load_model()
|
| 26 |
vision_tower.to(device=device, dtype=torch.float16)
|