From 49e4b57f70d247ac1aea248c245c85f767f57183 Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Tue, 29 Oct 2024 18:38:45 +0900 Subject: [PATCH] [benchmark] fix litgpt config import (#1346) --- thunder/benchmarks/distributed.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/thunder/benchmarks/distributed.py b/thunder/benchmarks/distributed.py index bc02528fe7..338ea91663 100644 --- 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