Skip to content

Commit 7c10568

Browse files
committed
fix llava kwargs in the hunyuan tests
1 parent 9677859 commit 7c10568

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/pipelines/hunyuan_video/test_hunyuan_image2video.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,9 @@ def get_dummy_components(self, num_layers: int = 1, num_single_layers: int = 1):
139139
num_hidden_layers=2,
140140
image_size=224,
141141
)
142-
llava_text_encoder_config = LlavaConfig(vision_config, text_config, pad_token_id=100, image_token_index=101)
142+
llava_text_encoder_config = LlavaConfig(
143+
vision_config=vision_config, text_config=text_config, pad_token_id=100, image_token_index=101
144+
)
143145

144146
clip_text_encoder_config = CLIPTextConfig(
145147
bos_token_id=0,

0 commit comments

Comments
 (0)