File tree Expand file tree Collapse file tree 2 files changed +2
-0
lines changed
swift/llm/infer/infer_engine Expand file tree Collapse file tree 2 files changed +2
-0
lines changed Original file line number Diff line number Diff line change 14
14
15
15
try :
16
16
# After setting the environment variables, import vllm. This way of writing allows lint to pass.
17
+ os .environ ['VLLM_USE_V1' ] = os .environ .get ('VLLM_USE_V1' , '0' )
17
18
os .environ ['VLLM_WORKER_MULTIPROC_METHOD' ] = 'spawn'
18
19
os .environ ['VLLM_ENGINE_ITERATION_TIMEOUT_S' ] = '3600'
19
20
import vllm
Original file line number Diff line number Diff line change 22
22
23
23
try :
24
24
# After setting the environment variables, import vllm. This way of writing allows lint to pass.
25
+ os .environ ['VLLM_USE_V1' ] = os .environ .get ('VLLM_USE_V1' , '0' )
25
26
os .environ ['VLLM_WORKER_MULTIPROC_METHOD' ] = 'spawn'
26
27
os .environ ['VLLM_ENGINE_ITERATION_TIMEOUT_S' ] = '3600'
27
28
import vllm
You can’t perform that action at this time.
0 commit comments