Fix

optimum/modeling_base.py

@@ -403,7 +403,21 @@ def from_pretrained(
                 trust_remote_code=trust_remote_code,
             )
 
-        from_pretrained_method = cls._export if export else cls._from_pretrained
+        if export:
+            if hasattr(cls, "_export"):
+                from_pretrained_method = cls._export
+            elif hasattr(cls, "_from_transformers"):
+                logger.warning(
+                    "The `export` argument is set to `True`, but the class does not implement `_export` methods. "
+                    "Using the `_from_transformers` method instead."
+                )
+                from_pretrained_method = cls._from_transformers
+            else:
+                raise ValueError(
+                    "The `export` argument is set to `True`, but the class does not implement `_export` methods."
+                )
+        else:
+            from_pretrained_method = cls._from_pretrained
 
         return from_pretrained_method(
             model_id=model_id,