diff --git a/superbench/benchmarks/model_benchmarks/megatron_gpt3.py b/superbench/benchmarks/model_benchmarks/megatron_gpt3.py index 379832b30..37d27bf1a 100644 --- a/superbench/benchmarks/model_benchmarks/megatron_gpt3.py +++ b/superbench/benchmarks/model_benchmarks/megatron_gpt3.py @@ -749,7 +749,7 @@ def _cal_params_count(self): 'megatron-deepseek-v2', MegatronGPT, parameters='--model=deepseek', platform=Platform.ROCM ) BenchmarkRegistry.register_benchmark( - 'megatron-deepseek-v2', + 'megatron-deepseek-v2-lite', MegatronGPT, parameters=( '--model=deepseek '