fix

modelscope · Aug 5, 2024 · 458168f · 458168f
1 parent 0ee10a3
commit 458168f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/swift/llm/utils/__init__.py b/swift/llm/utils/__init__.py
@@ -43,6 +43,8 @@
 try:
     if is_lmdeploy_available():
         from .lmdeploy_utils import (
+            lmdeploy_context,
+            LmdeployGenerationConfig,
             LmdeployGenerationConfig,
             get_lmdeploy_engine,
             inference_stream_lmdeploy,