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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-12-11 03:31:41 +00:00
commit cd0ca56eef
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
unsloth/models/llama.py
View file

@ -3011,6 +3011,7 @@ class FastLlamaModel:
output_embeddings = model.get_output_embeddings()
if input_embeddings is not None and output_embeddings is not None:
def _retie_parameter(target_module, source_module):
if not hasattr(source_module, "weight"):
return
@ -3029,9 +3030,9 @@ class FastLlamaModel:
if hasattr(input_embeddings, "modules_to_save") and hasattr(
output_embeddings, "modules_to_save"
):
if hasattr(input_embeddings.modules_to_save, "default") and hasattr(
output_embeddings.modules_to_save, "default"
):
if hasattr(
input_embeddings.modules_to_save, "default"
) and hasattr(output_embeddings.modules_to_save, "default"):
_retie_parameter(
output_embeddings.modules_to_save.default,
input_embeddings.modules_to_save.default,