Skip to content

Commit dbfbb59

Browse files
committed
[test] small fix to use vocab_size=1024
1 parent cf61b0a commit dbfbb59

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/pipelines/joyimage/test_joyimage_edit.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,11 +92,13 @@ def get_dummy_components(self):
9292

9393
scheduler = FlowMatchEulerDiscreteScheduler()
9494

95-
text_encoder = Qwen3VLForConditionalGeneration.from_pretrained(tiny_ckpt_id)
9695
processor = Qwen3VLProcessor.from_pretrained(tiny_ckpt_id)
9796
processor.image_processor.min_pixels = 4 * 28 * 28
9897
processor.image_processor.max_pixels = 4 * 28 * 28
9998

99+
text_encoder = Qwen3VLForConditionalGeneration.from_pretrained(tiny_ckpt_id)
100+
text_encoder.resize_token_embeddings(len(processor.tokenizer))
101+
100102
components = {
101103
"transformer": transformer,
102104
"vae": vae,

0 commit comments

Comments
 (0)