Spaces:
Runtime error
Runtime error
Update llava/model/builder.py
Browse files- llava/model/builder.py +1 -1
llava/model/builder.py
CHANGED
@@ -137,7 +137,7 @@ def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, l
|
|
137 |
|
138 |
|
139 |
vision_tower = model.get_vision_tower()
|
140 |
-
|
141 |
if not vision_tower.is_loaded:
|
142 |
vision_tower.load_model()
|
143 |
vision_tower.to(device=device, dtype=torch.float16)
|
|
|
137 |
|
138 |
|
139 |
vision_tower = model.get_vision_tower()
|
140 |
+
|
141 |
if not vision_tower.is_loaded:
|
142 |
vision_tower.load_model()
|
143 |
vision_tower.to(device=device, dtype=torch.float16)
|