Skip to content

Commit b799f4b

Browse files
authored
[CI/Build] Fix tensorizer test for load_format change (#22583)
Signed-off-by: 22quinn <[email protected]>
1 parent 06da44f commit b799f4b

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -426,7 +426,6 @@ steps:
426426

427427
- label: Tensorizer Test # 11min
428428
mirror_hardwares: [amdexperimental]
429-
soft_fail: true
430429
source_file_dependencies:
431430
- vllm/model_executor/model_loader
432431
- tests/tensorizer_loader

tests/entrypoints/openai/test_tensorizer_entrypoint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def model_uri(tmp_dir):
4444
def tensorize_model_and_lora(tmp_dir, model_uri):
4545
tensorizer_config = TensorizerConfig(tensorizer_uri=model_uri,
4646
lora_dir=tmp_dir)
47-
args = EngineArgs(model=MODEL_NAME, device="cuda")
47+
args = EngineArgs(model=MODEL_NAME)
4848

4949
tensorize_lora_adapter(LORA_PATH, tensorizer_config)
5050
tensorize_vllm_model(args, tensorizer_config)

tests/tensorizer_loader/test_tensorizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ def test_load_without_tensorizer_load_format(vllm_runner, capfd, model_ref):
166166
combined_output = out + err
167167
assert ("ValueError: Model loader extra config "
168168
"is not supported for load "
169-
"format LoadFormat.AUTO") in combined_output
169+
"format auto") in combined_output
170170
finally:
171171
del model
172172
gc.collect()
@@ -186,7 +186,7 @@ def test_raise_value_error_on_invalid_load_format(vllm_runner, capfd,
186186

187187
combined_output = out + err
188188
assert ("ValueError: Model loader extra config is not supported "
189-
"for load format LoadFormat.SAFETENSORS") in combined_output
189+
"for load format safetensors") in combined_output
190190
finally:
191191
del model
192192
gc.collect()

0 commit comments

Comments
 (0)