Skip to content

Commit 4eabf8f

Browse files
authored
Merge pull request #32 from huggingface/remove-warning
Remove tied weights warning
2 parents 6d564d0 + ff1df03 commit 4eabf8f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/transformers/models/llama4/configuration_llama4.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -335,6 +335,7 @@ def __init__(
335335
boi_token_index=200080,
336336
eoi_token_index=200081,
337337
image_token_index=200092,
338+
tie_word_embeddings = False,
338339
**kwargs,
339340
):
340341
if vision_config is None:
@@ -357,7 +358,7 @@ def __init__(
357358
elif isinstance(text_config, Llama4TextConfig):
358359
self.text_config = text_config
359360

360-
super().__init__(**kwargs)
361+
super().__init__(tie_word_embeddings=tie_word_embeddings, **kwargs)
361362

362363

363364
__all__ = ["Llama4Config", "Llama4TextConfig", "Llama4VisionConfig"]

0 commit comments

Comments
 (0)