Skip to content

Commit

Permalink
Update transformer.py
Browse files Browse the repository at this point in the history
Proposed fix for issue mistralai#225
  • Loading branch information
siwer authored Sep 27, 2024
1 parent db6b422 commit fc0e38f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/mistral_inference/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ def forward_partial(

if self.pipeline_rank == 0:
assert self.tok_embeddings is not None
if self.vision_encoder is not None and images:
if self.vision_encoder is not None and images is not None:
h = self.embed_vision_language_features(input_ids, images)
else:
h = self.tok_embeddings(input_ids)
Expand Down

0 comments on commit fc0e38f

Please sign in to comment.