diff --git a/wan/modules/model.py b/wan/modules/model.py
index 4478c7b..84f8dba 100644
--- a/wan/modules/model.py
+++ b/wan/modules/model.py
@@ -944,6 +944,7 @@ class WanModel(ModelMixin, ConfigMixin):
         if torch.is_tensor(freqs) and freqs.device != device:
             freqs = freqs.to(device)
 
+
         x_list = x
         joint_pass = len(x_list) > 1
         is_source_x = [ x.data_ptr() == x_list[0].data_ptr() and i > 0 for i, x in enumerate(x_list) ]
diff --git a/wgp.py b/wgp.py
index 5a0281b..b72469a 100644
--- a/wgp.py
+++ b/wgp.py
@@ -6295,10 +6295,10 @@ def set_new_tab(tab_state, new_tab_no):
         else:
             vmc_event_handler(True)
     tab_state["tab_no"] = new_tab_no
+    return gr.Tabs() 
 
 def select_tab(tab_state, evt:gr.SelectData):
-    set_new_tab(tab_state, evt.index)
-    return gr.Tabs() 
+    return set_new_tab(tab_state, evt.index)
 
 def get_js():
     start_quit_timer_js = """