Skip to content

Commit 29cba58

Browse files
authored
fix rollout(#4055)
1 parent ee831f5 commit 29cba58

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/infer/rollout.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def get_infer_engine(args: InferArguments, **kwargs):
131131
})
132132
infer_backend = kwargs.pop('infer_backend', None) or args.infer_backend
133133
if infer_backend != 'vllm':
134-
infer_backend == 'vllm'
134+
infer_backend = 'vllm'
135135
logger.info('Currently, rollout only supports the vLLM backend. Set vLLM backend')
136136
use_async_engine = args.use_async_engine
137137
if use_async_engine:

0 commit comments

Comments
 (0)