423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-29 18:20:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

[model] pushing FFT with unsloth (#8325)

Browse Source 
Co-authored-by: viyer <vivek_iyer2@apple.com>
This commit is contained in:
Vivek Iyer
2025-06-06 20:20:58 -04:00
committed by GitHub
parent 03a93ec513
commit 32b4574094
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/model/loader.py
View File

@@ -138,7 +138,7 @@ def load_model(
if model_args.adapter_name_or_path is not None:
lazy_load = True
elif is_trainable:
model = load_unsloth_pretrained_model(config, model_args)
model = load_unsloth_pretrained_model(config, model_args, finetuning_args)
if model is None and not lazy_load:
init_kwargs["config"] = config