423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-06-21 06:38:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] lint (#9636)

Browse Source
This commit is contained in:
Yaowei Zheng
2025-12-20 16:19:39 +08:00
committed by GitHub
parent b0d49e137f
commit 0894b4f37e
6 changed files with 28 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/llamafactory/train/dpo/workflow.py
View File

@@ -61,14 +61,14 @@ def run_dpo(
ref_model = create_ref_model(model_args, finetuning_args)
else:
ref_model = None
if model_args.use_kt:
from ktransformers.util.globals import GLOBAL_CONFIG
from ktransformers.util.globals import GLOBAL_CONFIG # type: ignore
from .ktrainer import KDPOTrainer as CustomDPOTrainer
GLOBAL_CONFIG._config["mod"] = "sft"
from .ktrainer import CustomDPOTrainer
else:
from .trainer import CustomDPOTrainer