Fix import of list_extended_configs in instantiator (#2285)

Summary: fyi xuzhao9 Pull Request resolved: #2285 Reviewed By: aaronenyeshi Differential Revision: D58369507 Pulled By: xuzhao9 fbshipit-source-id: 8f9519425fba6a015b1bca888a0e11492d62b1b6
pytorch · Jun 10, 2024 · e08d888 · e08d888
1 parent 37256de
commit e08d888
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchbenchmark/__init__.py b/torchbenchmark/__init__.py
@@ -615,7 +615,7 @@ def load_model_by_name(model_name: str):
     cls_name = "Model"
     if not models:
         # If the model is in TIMM or Huggingface extended model list
-        from torchbenchmark.util.framework.huggingface.extended_configs import (
+        from torchbenchmark.util.framework.huggingface.list_extended_configs import (
             list_extended_huggingface_models,
         )
         from torchbenchmark.util.framework.timm.extended_configs import (

diff --git a/torchbenchmark/util/experiment/instantiator.py b/torchbenchmark/util/experiment/instantiator.py
@@ -140,7 +140,7 @@ def list_models(internal=True) -> List[str]:
 
 
 def list_extended_models(suite_name: str = "all") -> List[str]:
-    from torchbenchmark.util.framework.huggingface.extended_configs import (
+    from torchbenchmark.util.framework.huggingface.list_extended_configs import (
         list_extended_huggingface_models,
     )
     from torchbenchmark.util.framework.timm.extended_configs import (