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 2025-03-09 23:03:27 -07:00
commit 186a2f6ee6
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/models/loader.py
View file

@ -108,8 +108,8 @@ class FastLanguageModel(FastLlamaModel):
use_gradient_checkpointing = use_gradient_checkpointing,
resize_model_vocab = resize_model_vocab, # [TODO] No effect
revision = revision,
return_logits = return_logits, # Return logits
fullgraph = fullgraph, # No graph breaks
return_logits = False, # Return logits
fullgraph = True, # No graph breaks
use_exact_model_name = use_exact_model_name,
*args, **kwargs,
)
@ -306,8 +306,8 @@ class FastLanguageModel(FastLlamaModel):
use_gradient_checkpointing = use_gradient_checkpointing,
resize_model_vocab = resize_model_vocab, # [TODO] No effect
revision = revision,
return_logits = return_logits, # Return logits
fullgraph = fullgraph, # No graph breaks
return_logits = False, # Return logits
fullgraph = True, # No graph breaks
use_exact_model_name = use_exact_model_name,
*args, **kwargs,
)