From bf3514a3d004fecad4664c5f93e917d13f08d610 Mon Sep 17 00:00:00 2001 From: Qing Lan Date: Thu, 8 Dec 2022 07:44:17 -0800 Subject: [PATCH] fix handler usage --- tests/integration/llm/prepare.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/integration/llm/prepare.py b/tests/integration/llm/prepare.py index 0b8196e8d..993e2125c 100644 --- a/tests/integration/llm/prepare.py +++ b/tests/integration/llm/prepare.py @@ -53,9 +53,9 @@ def build_hf_handler_model(model): def build_ds_handler_model(model): - if model not in hf_handler_list: - raise ValueError(f"{model} is not one of the supporting handler {list(hf_handler_list.keys())}") - options = hf_handler_list[model] + if model not in ds_handler_list: + raise ValueError(f"{model} is not one of the supporting handler {list(ds_handler_list.keys())}") + options = ds_handler_list[model] options["engine"] = "DeepSpeed" options["option.entryPoint"] = "djl_python.deepspeed" write_prperties(options)