Skip to content

Commit db3d794

Browse files
authored
Removed explicit mode for multi-lora (#45)
1 parent 18a96e3 commit db3d794

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

ci/L0_multi_gpu/multi_lora/multi_lora_test.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,6 @@ def _test_vllm_model(
119119
self.triton_client.stop_stream()
120120

121121
def test_multi_lora_requests(self):
122-
self.triton_client.load_model(self.vllm_model_name)
123122
sampling_parameters = {"temperature": "0", "top_p": "1"}
124123
# make two requests separately to avoid the different arrival of response answers
125124
prompt_1 = ["Instruct: What do you think of Computer Science?\nOutput:"]
@@ -151,10 +150,8 @@ def test_multi_lora_requests(self):
151150
exclude_input_in_output=True,
152151
expected_output=expected_output,
153152
)
154-
self.triton_client.unload_model(self.vllm_model_name)
155153

156154
def test_none_exist_lora(self):
157-
self.triton_client.load_model(self.vllm_model_name)
158155
prompts = [
159156
"Instruct: What is the capital city of France?\nOutput:",
160157
]
@@ -169,7 +166,6 @@ def test_none_exist_lora(self):
169166
exclude_input_in_output=True,
170167
expected_output=None, # this request will lead to lora not supported error, so there is no expected output
171168
)
172-
self.triton_client.unload_model(self.vllm_model_name)
173169

174170
def tearDown(self):
175171
self.triton_client.close()

ci/L0_multi_gpu/multi_lora/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ source ../../common/util.sh
3030
TRITON_DIR=${TRITON_DIR:="/opt/tritonserver"}
3131
SERVER=${TRITON_DIR}/bin/tritonserver
3232
BACKEND_DIR=${TRITON_DIR}/backends
33-
SERVER_ARGS="--model-repository=`pwd`/models --backend-directory=${BACKEND_DIR} --model-control-mode=explicit --log-verbose=1"
33+
SERVER_ARGS="--model-repository=`pwd`/models --backend-directory=${BACKEND_DIR} --log-verbose=1"
3434
SERVER_LOG="./multi_lora_server.log"
3535
CLIENT_LOG="./multi_lora_client.log"
3636
TEST_RESULT_FILE='test_results.txt'

0 commit comments

Comments
 (0)