From aba3002b83f02b200e1bce786e4d531705629bbb Mon Sep 17 00:00:00 2001
From: Chris Malone <chrismrulz@hotmail.com>
Date: Sun, 6 Apr 2025 04:32:57 +1000
Subject: [PATCH] fix gen referenced before assignment bug when switching tabs

---
 gradio_server.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/gradio_server.py b/gradio_server.py
index 366cf39..883f7e7 100644
--- a/gradio_server.py
+++ b/gradio_server.py
@@ -3240,11 +3240,18 @@ def on_tab_select(global_state, t2v_state, i2v_state, evt: gr.SelectData):
 
         global_state["last_tab_was_image2video"] = new_i2v
 
-    if(server_config.get("reload_model",2) == 1):
+    if server_config.get("reload_model", 2) == 1:
         queue = gen.get("queue", [])
+        queue_empty = len(queue) == 0
 
-        queue_empty = len(queue) == 0    
-        if queue_empty:
+        is_switching_between_gen_tabs = (
+            last_tab_was_image2video is not None and
+            (new_t2v or new_i2v) and
+            last_tab_was_image2video != new_i2v
+        )
+
+        if is_switching_between_gen_tabs and queue_empty:
+            print("Reloading model due to switch between T2V/I2V tabs.")
             global wan_model, offloadobj
             if wan_model is not None:
                 if offloadobj is not None: