Fix failing tests for cohere_compass (#48005)
* fix(cohere_compass): address failing tests
Fixed 8 test(s):
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_model_parallel_beam_search
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_model_parallelism
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_torch_export_dynamic
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_torch_export_generate_dynamic
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_torch_export_generate_dynamic_static_cache
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_torch_export_generate_static
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_torch_export_generate_static_static_cache
- tests/models/cohere_compass/test_modeling_cohere_compass.py::CohereCompassModelTest::test_torch_export_static
* update
Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>
* update
Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>
* update
Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>
---------
Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>
Co-authored-by: Raushan Turganbay <raushan@huggingface.co>
