fix warnings.warn (#111)

ad1a8638 · A60 · GitHub · c0933457 · ad1a8638
Unverified Commit ad1a8638 authored Jul 14, 2023 by A60 Committed by GitHub Jul 14, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lmdeploy/pytorch/chat.py lmdeploy/pytorch/chat.py +1 -1

No files found.
--- a/lmdeploy/pytorch/chat.py
+++ b/lmdeploy/pytorch/chat.py
@@ -73,7 +73,7 @@ def init_model(
    model = model.cuda(local_rank)

    if not _is_deepspeed_available:
-        warnings.warn('deepspeed is not installed, ',
+        warnings.warn('deepspeed is not installed, '
                      'use plain huggingface model.')
    else:
        model = deepspeed.init_inference(