VitekDev/ai-voice-cloning
1
0
Fork 0
forked from mrq/ai-voice-cloning
Code Pull Requests Activity

small QoL change to the StringNone helper, to allow generated text to be "None", maybe someone wants to generate that, we never know...

Browse Source
This commit is contained in:
ben_mkiv 2023-08-22 21:49:49 +02:00
parent 9abcb0f193
commit 5d73d9e71c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/api/websocket_server.py
View File

@ -9,8 +9,10 @@ from utils import generate, get_autoregressive_models, get_voice_list
# this is a not so nice workaround to set values to None if their string value is "None" # this is a not so nice workaround to set values to None if their string value is "None"
def replaceNoneStringWithNone(message): def replaceNoneStringWithNone(message):
ignore_fields = ['text'] # list of fields which CAN have "None" as literal String value
for member in message: for member in message:
if message[member] == 'None': if message[member] == 'None' and member not in ignore_fields:
message[member] = None message[member] = None
return message return message