Make FeaturesManager.get_model_from_feature a static method (huggingf…

…ace#16357)
pathwaymd · Mar 25, 2022 · 2b23e08 · 2b23e08
1 parent aa6cfe9
commit 2b23e08
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/onnx/features.py b/src/transformers/onnx/features.py
@@ -344,6 +344,7 @@ def get_model_class_for_feature(feature: str, framework: str = "pt") -> Type:
             )
         return task_to_automodel[task]
 
+    @staticmethod
     def get_model_from_feature(
         feature: str, model: str, framework: str = "pt", cache_dir: str = None
     ) -> Union[PreTrainedModel, TFPreTrainedModel]: