Spaces:
Running
on
Zero
Running
on
Zero
Update oryx/model/builder.py
Browse files- oryx/model/builder.py +1 -1
oryx/model/builder.py
CHANGED
|
@@ -72,7 +72,7 @@ def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, l
|
|
| 72 |
model.resize_token_embeddings(len(tokenizer))
|
| 73 |
|
| 74 |
vision_tower = model.get_vision_tower()
|
| 75 |
-
print("Loading vision tower
|
| 76 |
if not vision_tower.is_loaded:
|
| 77 |
vision_tower.load_model(device_map=device_map)
|
| 78 |
# if device_map != "auto":
|
|
|
|
| 72 |
model.resize_token_embeddings(len(tokenizer))
|
| 73 |
|
| 74 |
vision_tower = model.get_vision_tower()
|
| 75 |
+
print("Loading vision tower....")
|
| 76 |
if not vision_tower.is_loaded:
|
| 77 |
vision_tower.load_model(device_map=device_map)
|
| 78 |
# if device_map != "auto":
|