do not untie

kylesayrs · kylesayrs · commit 8b8140a43ad2 · 2025-07-30T17:45:51.000-04:00
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/examples/multimodal_vision/gemma3n_example.py b/examples/multimodal_vision/gemma3n_example.py
@@ -55,6 +55,8 @@ def data_collator(batch):
     data_collator=data_collator,
     # gemma3n has broken weight offloading which is required by the sequential pipeline
     pipeline="basic",
+    # gemma3n does not support untying word embeddings
+    tie_word_embeddings=True,
 )
 
 # Confirm generations of the quantized model look sane.

Original file line number	Diff line number	Diff line change
`@@ -55,6 +55,8 @@ def data_collator(batch):`
`55`	`55`	`data_collator=data_collator,`
`56`	`56`	`# gemma3n has broken weight offloading which is required by the sequential pipeline`
`57`	`57`	`pipeline="basic",`
	`58`	`+ # gemma3n does not support untying word embeddings`
	`59`	`+ tie_word_embeddings=True,`
`58`	`60`	`)`
`59`	`61`
`60`	`62`	`# Confirm generations of the quantized model look sane.`