Spaces:
Runtime error
Runtime error
Commit
•
8130fc5
1
Parent(s):
43d70a9
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ model = LlavaLlamaForCausalLM.from_pretrained('HuanjinYao/DenseConnector-v1.5-8B
|
|
23 |
vision_tower = model.get_vision_tower()
|
24 |
if not vision_tower.is_loaded:
|
25 |
vision_tower.load_model()
|
26 |
-
vision_tower.to(device=
|
27 |
image_processor = vision_tower.image_processor
|
28 |
|
29 |
model.to('cuda')
|
|
|
23 |
vision_tower = model.get_vision_tower()
|
24 |
if not vision_tower.is_loaded:
|
25 |
vision_tower.load_model()
|
26 |
+
vision_tower.to(device='cuda', dtype=torch.float16)
|
27 |
image_processor = vision_tower.image_processor
|
28 |
|
29 |
model.to('cuda')
|