mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-04 04:32:50 +08:00
Update adapter.py
Former-commit-id: 98bc97d8d218182c026e9f57bbcbf40ab1e0bc87
This commit is contained in:
parent
e5b4cb62e0
commit
77d16ada1e
@ -145,7 +145,7 @@ def init_adapter(
|
||||
if model_args.use_unsloth:
|
||||
from unsloth import FastLanguageModel # type: ignore
|
||||
|
||||
unsloth_peft_kwargs = {"model": model, "max_seq_length": model_args.model_max_length}
|
||||
unsloth_peft_kwargs = {"model": model, "max_seq_length": model_args.model_max_length}
|
||||
model = FastLanguageModel.get_peft_model(**peft_kwargs, **unsloth_peft_kwargs)
|
||||
else:
|
||||
lora_config = LoraConfig(
|
||||
|
Loading…
x
Reference in New Issue
Block a user