Skip to content

Commit

Permalink
fix after rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
eaidova committed Nov 14, 2024
1 parent cf148c2 commit 6108184
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions optimum/intel/openvino/modeling_visual_language.py
Original file line number Diff line number Diff line change
Expand Up @@ -261,12 +261,11 @@ def forward(self, img_features):
return self.request(img_features)[0]



MODEL_PARTS_CLS_MAPPING = {
"resampler": OVResampler,
"language_model": OVModelWithEmbedForCausalLM,
"vision_embeddings": OVVisionEmbedding,
"vision_projection": OVVisionProjection
"vision_projection": OVVisionProjection,
}


Expand Down

0 comments on commit 6108184

Please sign in to comment.