support export size setting

Former-commit-id: 1a4de54586c21cdbbc89f8a716ca5a54c87a6120
2023-11-26 18:34:09 +08:00
parent 953a562ec1
commit 569860d7ac
4 changed files with 11 additions and 5 deletions
--- a/src/llmtuner/webui/chatter.py
+++ b/src/llmtuner/webui/chatter.py
@@ -90,6 +90,7 @@ class WebChatModel(ChatModel):
        lang = data[self.manager.get_elem_by_name("top.lang")]

        if self.demo_mode:
+            gr.Warning(ALERTS["err_demo"][lang])
            yield ALERTS["err_demo"][lang]
            return