File tree Expand file tree Collapse file tree 2 files changed +1
-5
lines changed
ci/L0_multi_gpu/multi_lora Expand file tree Collapse file tree 2 files changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -119,7 +119,6 @@ def _test_vllm_model(
119
119
self .triton_client .stop_stream ()
120
120
121
121
def test_multi_lora_requests (self ):
122
- self .triton_client .load_model (self .vllm_model_name )
123
122
sampling_parameters = {"temperature" : "0" , "top_p" : "1" }
124
123
# make two requests separately to avoid the different arrival of response answers
125
124
prompt_1 = ["Instruct: What do you think of Computer Science?\n Output:" ]
@@ -151,10 +150,8 @@ def test_multi_lora_requests(self):
151
150
exclude_input_in_output = True ,
152
151
expected_output = expected_output ,
153
152
)
154
- self .triton_client .unload_model (self .vllm_model_name )
155
153
156
154
def test_none_exist_lora (self ):
157
- self .triton_client .load_model (self .vllm_model_name )
158
155
prompts = [
159
156
"Instruct: What is the capital city of France?\n Output:" ,
160
157
]
@@ -169,7 +166,6 @@ def test_none_exist_lora(self):
169
166
exclude_input_in_output = True ,
170
167
expected_output = None , # this request will lead to lora not supported error, so there is no expected output
171
168
)
172
- self .triton_client .unload_model (self .vllm_model_name )
173
169
174
170
def tearDown (self ):
175
171
self .triton_client .close ()
Original file line number Diff line number Diff line change @@ -30,7 +30,7 @@ source ../../common/util.sh
30
30
TRITON_DIR=${TRITON_DIR:= " /opt/tritonserver" }
31
31
SERVER=${TRITON_DIR} /bin/tritonserver
32
32
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"
34
34
SERVER_LOG=" ./multi_lora_server.log"
35
35
CLIENT_LOG=" ./multi_lora_client.log"
36
36
TEST_RESULT_FILE=' test_results.txt'
You can’t perform that action at this time.
0 commit comments