Skip to content

Commit

Permalink
Update adapter.py
Browse files Browse the repository at this point in the history
  • Loading branch information
hiyouga authored Apr 9, 2024
1 parent 2111b58 commit 98bc97d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/llmtuner/model/adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 98bc97d

Please sign in to comment.