Update llama.py
This commit is contained in:
parent
b637868fb2
commit
44f405a009
1 changed files with 1 additions and 1 deletions
|
|
@ -2019,7 +2019,7 @@ class FastLlamaModel:
|
|||
**kwargs,
|
||||
):
|
||||
if os.environ.get("UNSLOTH_USE_NEW_MODEL", "0") == "1":
|
||||
return FastBaseModel.get_model(
|
||||
return FastBaseModel.get_peft_model(
|
||||
model = model,
|
||||
r = r,
|
||||
target_modules = target_modules,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue