Merge pull request #1067 from Ikaros-521/owner

修复：GSV删除ws配置导致的运行bug
Ikaros-521 · Dec 12, 2024 · 9b95609 · 9b95609
2 parents 5824a06 + 9d09836
commit 9b95609
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/main.py b/main.py
@@ -2446,7 +2446,7 @@ def on_message(ws, message):
         def on_error(ws, error):
             logger.error(f"Error:{error}")
 
-        def on_close(ws):
+        def on_close(ws, close_status_code, close_msg):
             logger.debug("WebSocket connection closed")
 
         def on_open(ws):

diff --git a/utils/audio.py b/utils/audio.py
@@ -1080,7 +1080,6 @@ async def tts_handle(self, message):
                 data = {
                     "type": message["data"]["type"],
                     "gradio_ip_port": message["data"]["gradio_ip_port"],
-                    "ws_ip_port": message["data"]["ws_ip_port"],
                     "api_ip_port": message["data"]["api_ip_port"],
                     "ref_audio_path": message["data"]["ref_audio_path"],
                     "prompt_text": message["data"]["prompt_text"],