Skip to content

Commit d955a79

Browse files
authored
fix bugs (#793)
1 parent 8528701 commit d955a79

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/infer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def llm_infer(args: InferArguments) -> None:
257257
if args.save_result:
258258
result_dir = args.ckpt_dir
259259
if result_dir is None:
260-
result_dir = model.model_dir
260+
result_dir = llm_engine.model_dir if args.infer_backend == 'vllm' else model.model_dir
261261
if result_dir is not None:
262262
result_dir = os.path.join(result_dir, 'infer_result')
263263
os.makedirs(result_dir, exist_ok=True)

0 commit comments

Comments
 (0)