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 (