test vocab size

linoytsaban · linoytsaban · commit de5fad17fd3d · 2025-05-26T13:26:09.000+03:00
diff --git a/tests/pipelines/sana/test_sana_sprint_img2img.py b/tests/pipelines/sana/test_sana_sprint_img2img.py
@@ -120,7 +120,7 @@ def get_dummy_components(self):
             num_attention_heads=2,
             num_hidden_layers=1,
             num_key_value_heads=2,
-            vocab_size=8,
+            vocab_size=1000,
             attn_implementation="eager",
         )
         text_encoder = Gemma2Model(text_encoder_config)

Original file line number	Diff line number	Diff line change
`@@ -120,7 +120,7 @@ def get_dummy_components(self):`
`120`	`120`	`num_attention_heads=2,`
`121`	`121`	`num_hidden_layers=1,`
`122`	`122`	`num_key_value_heads=2,`
`123`		`- vocab_size=8,`
	`123`	`+ vocab_size=1000,`
`124`	`124`	`attn_implementation="eager",`
`125`	`125`	`)`
`126`	`126`	`text_encoder = Gemma2Model(text_encoder_config)`