From 9236260a4a3f85dbc013958b660ca56d7093dec2 Mon Sep 17 00:00:00 2001
From: jingyaogong <gongjy.cs@qq.com>
Date: Wed, 31 Dec 2025 20:46:59 +0800
Subject: [PATCH] [feat] get params

---
 trainer/trainer_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trainer/trainer_utils.py b/trainer/trainer_utils.py
index 49f263f..a50737b 100644
--- a/trainer/trainer_utils.py
+++ b/trainer/trainer_utils.py
@@ -12,7 +12,7 @@ import torch
 import torch.distributed as dist
 from torch.utils.data import Sampler
 from transformers import AutoTokenizer
-from model.model_minimind_qwen3 import MiniMindForCausalLM
+from model.model_minimind import MiniMindForCausalLM
 
 def get_model_params(model, config):
     total = sum(p.numel() for p in model.parameters()) / 1e6