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( # 原始模型