From e08d888dee37f15b8c12adc28461e5936af54396 Mon Sep 17 00:00:00 2001 From: Thomas Roehr Date: Mon, 10 Jun 2024 12:14:34 -0700 Subject: [PATCH] Fix import of list_extended_configs in instantiator (#2285) Summary: fyi xuzhao9 Pull Request resolved: https://github.com/pytorch/benchmark/pull/2285 Reviewed By: aaronenyeshi Differential Revision: D58369507 Pulled By: xuzhao9 fbshipit-source-id: 8f9519425fba6a015b1bca888a0e11492d62b1b6 --- torchbenchmark/__init__.py | 2 +- torchbenchmark/util/experiment/instantiator.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchbenchmark/__init__.py b/torchbenchmark/__init__.py index 07f0e9be2f..142b547f4c 100644 --- 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 index a6acc0ec31..08d13d9cc2 100644 --- 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 (