Make FeaturesManager.get_model_from_feature a static method (#16357)

2b23e080 · Michael Benayoun · GitHub · aa6cfe9c · 2b23e080
Unverified Commit 2b23e080 authored Mar 25, 2022 by Michael Benayoun Committed by GitHub Mar 25, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/onnx/features.py src/transformers/onnx/features.py +1 -0

No files found.
--- a/src/transformers/onnx/features.py
+++ b/src/transformers/onnx/features.py
@@ -344,6 +344,7 @@ class FeaturesManager:
            )
        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]: