fix litgpt config import

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
Lightning-AI · Oct 24, 2024 · ed5278d · ed5278d
1 parent 79e59d0
commit ed5278d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/thunder/benchmarks/distributed.py b/thunder/benchmarks/distributed.py
@@ -12,9 +12,8 @@
     NanoGPTConfig,
     NanoGPTBenchmark,
     LitGPTBenchmark,
-    LitGPTConfig,
 )
-from thunder.tests.litgpt_model import name_to_config
+from thunder.tests.litgpt_model import name_to_config, Config as LitGPTConfig
 from thunder.distributed import FSDPBucketingStrategy
 from thunder.distributed import FSDPType