ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update loader.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-31 03:46:11 +11:00
commit d347db0944
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/loader.py
View file

@ -71,7 +71,7 @@ class FastLanguageModel(FastLlamaModel):
device_map = "sequential",
rope_scaling = None,
fix_tokenizer = True,
use_gradient_checkpointing = False,
use_gradient_checkpointing = True,
*args, **kwargs,
):
old_model_name = model_name