diff --git a/app_tests/integration_tests/llm/shortfin/conftest.py b/app_tests/integration_tests/llm/shortfin/conftest.py index 438914b31..f6a23960a 100644 --- a/app_tests/integration_tests/llm/shortfin/conftest.py +++ b/app_tests/integration_tests/llm/shortfin/conftest.py @@ -87,7 +87,7 @@ def model_test_dir(request, tmp_path_factory): "paged_kv_cache": { "block_seq_stride": 16, "device_block_count": 256, - "prefix_sharing_algorithm": "none", + "prefix_sharing_algorithm": prefix_sharing_algorithm, }, } logger.info(f"Saving edited config to: {edited_config_path}\n")