From 0ed43a963c280a3c0beef7ce57b129854aaeb8bc Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Sun, 27 Apr 2025 22:58:46 +0800 Subject: [PATCH] =?UTF-8?q?refactor(train):=20=E7=A7=BB=E9=99=A4=E6=9C=AA?= =?UTF-8?q?=E4=BD=BF=E7=94=A8=E7=9A=84=E6=A8=A1=E5=9E=8B=E9=85=8D=E7=BD=AE?= =?UTF-8?q?=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 移除未使用的 `dtype` 和 `load_in_4bit` 变量以简化代码结构 --- train/model.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/train/model.py b/train/model.py index 2b70c48..e05b1df 100644 --- a/train/model.py +++ b/train/model.py @@ -43,9 +43,6 @@ def train_model( lora_rank: int, trainer_callback=None ) -> None: - # 模型配置参数 - dtype = None # 数据类型,None表示自动选择 - load_in_4bit = False # 使用4bit量化加载模型以节省显存 model = FastLanguageModel.get_peft_model( # 原始模型