Skip to content

Commit 008de41

Browse files
Merge pull request #44 from RadoslawPlawecki/solve-issue-39
Resolve the issue of getting different results from transformers evaluation and vLLM evaluation
2 parents 70489db + a51a18e commit 008de41

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

llmsql/inference/inference_vllm.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@
2222
"""
2323

2424
import os
25+
26+
os.environ["VLLM_USE_V1"] = "0"
27+
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
28+
os.environ["VLLM_ENABLE_V1_MULTIPROCESSING"] = "0"
29+
2530
from pathlib import Path
2631
from typing import Any
2732

0 commit comments

Comments
 (0)