bowong
/
Wan2GP
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix gen referenced before assignment bug when switching tabs

This commit is contained in:
Chris Malone 2025-04-06 04:32:57 +10:00
parent 3b066849de
commit aba3002b83
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
gradio_server.py
View File

@ -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 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 = 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 global wan_model, offloadobj
if wan_model is not None: if wan_model is not None:
if offloadobj is not None: if offloadobj is not None: