Skip to content

Commit ebb1f0f

Browse files
authored
model: fix colpali for update transformer (#314)
1 parent 781639e commit ebb1f0f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/optimum/rbln/transformers/models/colpali/modeling_colpali.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def from_model(
270270
"""
271271
if not hasattr(model, "vision_tower"):
272272
model.vision_tower = model.vlm.vision_tower
273-
del model.vlm.vision_tower
273+
del model.vlm.model.vision_tower
274274
model = super().from_model(model, config, rbln_config, model_save_dir, subfolder, **kwargs)
275275
return model
276276

0 commit comments

Comments
 (0)