diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 2d14e6364f7..3dfa786dc92 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -2994,7 +2994,7 @@ else: "Pix2StructVisionModel", ] ) - _import_structure["models.pixtral"].extend(["PixtralVisionModel", "PixtralPreTrainedModel"]) + _import_structure["models.pixtral"].extend(["PixtralPreTrainedModel", "PixtralVisionModel"]) _import_structure["models.plbart"].extend( [ "PLBartForCausalLM",