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(
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?\n Output:" ]
@@ -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?\n Output:" ,
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 ()
Original file line number Diff line number Diff line change @@ -30,7 +30,7 @@ source ../../common/util.sh
3030TRITON_DIR=${TRITON_DIR:= " /opt/tritonserver" }
3131SERVER=${TRITON_DIR} /bin/tritonserver
3232BACKEND_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"
3434SERVER_LOG=" ./multi_lora_server.log"
3535CLIENT_LOG=" ./multi_lora_client.log"
3636TEST_RESULT_FILE=' test_results.txt'
You can’t perform that action at this time.
0 commit comments