Skip to content

GitLab

"tools/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "6013e23c6dd4202b02470dd22643fa058f369f54"
Unverified Commit db3b986b authored by Kevin Wang's avatar Kevin Wang Committed by GitHub
Browse files

[Fix] fix attempted_relative_import (#125) 

* [Fix] fix attempted_relative_import

* use try...except...else
parent 9eca5cbe
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
lmdeploy/pytorch/chat.py
View file @ db3b986b
...@@ -17,12 +17,12 @@ try: ...@@ -17,12 +17,12 @@ try:
from transformers import (AutoModelForCausalLM, AutoTokenizer, from transformers import (AutoModelForCausalLM, AutoTokenizer,
GenerationConfig) GenerationConfig)
from .accel import LoadNoInit
from .utils import get_utils
_is_transformers_available = True _is_transformers_available = True
except ImportError: except ImportError:
_is_transformers_available = False _is_transformers_available = False
else:
from .accel import LoadNoInit
from .utils import get_utils
def input_prompt(): def input_prompt():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment