Skip to content

Commit 09e6cfe

Browse files
committed
add gemma3n vision example
Signed-off-by: Kyle Sayers <[email protected]>
1 parent d69da65 commit 09e6cfe

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/multimodal_vision/gemma3n_example.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,8 @@ def data_collator(batch):
5353
num_calibration_samples=NUM_CALIBRATION_SAMPLES,
5454
trust_remote_code_model=True,
5555
data_collator=data_collator,
56+
# gemma3n has broken weight offloading which is required by the sequential pipeline
57+
pipeline="basic",
5658
)
5759

5860
# Confirm generations of the quantized model look sane.

0 commit comments

Comments
 (0)