fix #3317

2024-04-17 22:17:19 +08:00
parent e3ca57602c
commit 6d641af703
15 changed files with 75 additions and 32 deletions
--- a/src/llmtuner/webui/components/chatbot.py
+++ b/src/llmtuner/webui/components/chatbot.py
@@ -1,11 +1,14 @@
 from typing import TYPE_CHECKING, Dict, Tuple

-import gradio as gr
-
 from ...data import Role
+from ...extras.packages import is_gradio_available
 from ..utils import check_json_schema


+if is_gradio_available():
+    import gradio as gr
+
+
 if TYPE_CHECKING:
    from gradio.components import Component