Unverified Commit b34129bf authored by WeiQing Chen's avatar WeiQing Chen Committed by GitHub
Browse files

[Misc] remove useless v1 env (#29164)


Signed-off-by: default avatarDavid Chen <530634352@qq.com>
parent 4d7231e7
...@@ -61,8 +61,6 @@ def test_batch_inference_correctness( ...@@ -61,8 +61,6 @@ def test_batch_inference_correctness(
model_setup: (method, model_name, spec_model_name, lora_path, tp_size) model_setup: (method, model_name, spec_model_name, lora_path, tp_size)
""" """
with monkeypatch.context() as m: with monkeypatch.context() as m:
m.setenv("VLLM_USE_V1", "1")
# Disable randomness # Disable randomness
m.setenv("CUBLAS_WORKSPACE_CONFIG", ":4096:8") m.setenv("CUBLAS_WORKSPACE_CONFIG", ":4096:8")
torch.manual_seed(SEED) torch.manual_seed(SEED)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment