Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions tests/ut/core/test_schedule_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ def setUp(self):
max_model_len=8192,
is_multimodal_model=False,
send_delta_data=False,
is_encoder_decoder=False,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

This correctly adds the missing is_encoder_decoder argument. However, there is another instantiation of SchedulerConfig in this file that also needs to be updated. On line 90, SchedulerConfig is called without is_encoder_decoder, which will likely cause the test_valid_config_with_multimodal test to fail with a pydantic.ValidationError.

Please update lines 90-91 from:

SchedulerConfig(is_multimodal_model=True,
                max_num_batched_tokens=8192), {})

to:

SchedulerConfig(is_multimodal_model=True,
                max_num_batched_tokens=8192,
                is_encoder_decoder=False), {})

)

def test_initialize_from_config_with_default(self):
Expand Down
Loading