[benchmark] fix litgpt config import (#1346)

Lightning-AI · Oct 29, 2024 · 49e4b57 · 49e4b57
1 parent a7618f6
commit 49e4b57
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