diff --git a/tests/ut/core/test_scheduler.py b/tests/ut/core/test_scheduler.py index 635e08d..5c793d1 100644 --- a/tests/ut/core/test_scheduler.py +++ b/tests/ut/core/test_scheduler.py @@ -135,6 +135,8 @@ class TestAscendScheduler(TestBase): ) model_config.pooler_config = MagicMock() model_config.multimodal_config = MagicMock() + model_config.hf_config = MagicMock() + model_config.hf_config.is_encoder_decoder = False # Cache config, optionally force APC kwargs_cache: Dict[str, Any] = ({} if ENABLE_PREFIX_CACHING is None else {