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

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2025-01-01 22:34:40 -08:00
commit bd506f2c44
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/_utils.py
View file

@ -561,7 +561,7 @@ def prepare_model_for_kbit_training(
model.get_input_embeddings().register_forward_hook(make_inputs_require_grad)
# Enable grads on non language models as well
requires_grad_for_gradient_checkpointing()
requires_grad_for_gradient_checkpointing(model)
pass
return model