[Fix] fix attempted_relative_import (#125)

* [Fix] fix attempted_relative_import * use try...except...else

[Fix] fix attempted_relative_import (#125)
* [Fix] fix attempted_relative_import * use try...except...else
db3b986b · Kevin Wang · GitHub · 9eca5cbe · db3b986b
Unverified Commit db3b986b authored Jul 17, 2023 by Kevin Wang Committed by GitHub Jul 17, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

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

No files found.
--- a/lmdeploy/pytorch/chat.py
+++ b/lmdeploy/pytorch/chat.py
@@ -17,12 +17,12 @@ try:
    from transformers import (AutoModelForCausalLM, AutoTokenizer,
                              GenerationConfig)

-    from .accel import LoadNoInit
-    from .utils import get_utils
-
    _is_transformers_available = True
 except ImportError:
    _is_transformers_available = False
+else:
+    from .accel import LoadNoInit
+    from .utils import get_utils


 def input_prompt():