fix webchat

Former-commit-id: 071fe40f209156f994c069507a2d53cc4f586d67
2024-11-02 21:04:18 +08:00
parent 2bb3255e74
commit c610deb5a2
2 changed files with 3 additions and 4 deletions
--- a/src/llamafactory/webui/chatter.py
+++ b/src/llamafactory/webui/chatter.py
@@ -144,8 +144,8 @@ class WebChatModel(ChatModel):
            messages,
            system,
            tools,
-            images=[image],
-            videos=[video],
+            images=[image] if image else None,
+            videos=[video] if video else None,
            max_new_tokens=max_new_tokens,
            top_p=top_p,
            temperature=temperature,