Remove model_architecture to get tiny_config.

huggingface · Dec 13, 2021 · d56ad2a · d56ad2a
1 parent 55d0061
commit d56ad2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_pipelines_common.py b/tests/test_pipelines_common.py
@@ -54,7 +54,7 @@ def get_checkpoint_from_architecture(architecture):
         logger.warning(f"Can't retrieve checkpoint from {architecture.__name__}")
 
 
-def get_tiny_config_from_class(configuration_class, model_architecture):
+def get_tiny_config_from_class(configuration_class):
     if "OpenAIGPT" in configuration_class.__name__:
         # This is the only file that is inconsistent with the naming scheme.
         # Will rename this file if we decide this is the way to go
@@ -221,7 +221,7 @@ def data(n):
 
                     for model_architecture in model_architectures:
                         checkpoint = get_checkpoint_from_architecture(model_architecture)
-                        tiny_config = get_tiny_config_from_class(configuration, model_architecture)
+                        tiny_config = get_tiny_config_from_class(configuration)
                         tokenizer_classes = TOKENIZER_MAPPING.get(configuration, [])
                         feature_extractor_class = FEATURE_EXTRACTOR_MAPPING.get(configuration, None)
                         feature_extractor_name = (