Apply Zero GPU for gr.ChatInterface (#10081)

* zero gpu * add changeset * changes * changes * add changeset * change * check * add changeset * format --------- Co-authored-by: gradio-pr-bot <[email protected]>
gradio-app · Dec 1, 2024 · b94f010 · b94f010
1 parent ebe25bc
commit b94f010
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.changeset/little-meals-enter.md b/.changeset/little-meals-enter.md
@@ -0,0 +1,5 @@
+---
+"gradio": minor
+---
+
+feat:Apply Zero GPU for `gr.ChatInterface`
diff --git a/gradio/chat_interface.py b/gradio/chat_interface.py
@@ -337,8 +337,10 @@ def _setup_example_messages(
         return examples_messages
 
     def _setup_events(self) -> None:
-        submit_fn = self._stream_fn if self.is_generator else self._submit_fn
         submit_triggers = [self.textbox.submit, self.chatbot.retry]
+        submit_fn = self._stream_fn if self.is_generator else self._submit_fn
+        if hasattr(self.fn, "zerogpu"):
+            submit_fn.__func__.zerogpu = self.fn.zerogpu  # type: ignore
 
         submit_event = (
             self.textbox.submit(