changes for blackbird - model sharding

Author: Ashish Gupta

tests/unit/sagemaker/serve/builder/test_model_builder.py

@@ -2679,6 +2679,7 @@ def test_optimize_exclusive_sharding(self, mock_get_serve_setting):
             ValueError,
             "Sharding config is mutually exclusive and cannot be combined with any other optimization.",
             lambda: model_builder.optimize(
+                quantization_config={"OverrideEnvironment": {"OPTION_QUANTIZE": "awq"}},
                 compilation_config={"OverrideEnvironment": {"OPTION_QUANTIZE": "awq"}},
                 sharding_config={"OverrideEnvironment": {"OPTION_QUANTIZE": "awq"}},
             ),