Websocket fixes / additions #350

Merged
mrq merged 4 commits from ben_mkiv/ai-voice-cloning:master into master 2023-08-30 18:39:35 +00:00
Showing only changes of commit 6f0f148782 - Show all commits

View File

@ -4,7 +4,7 @@ from threading import Thread
from websockets.server import serve
from utils import generate, get_autoregressive_models, get_voice_list, args
from utils import generate, get_autoregressive_models, get_voice_list, args, update_autoregressive_model, update_diffusion_model, update_tokenizer
# this is a not so nice workaround to set values to None if their string value is "None"
def replaceNoneStringWithNone(message):
@ -18,19 +18,18 @@ def replaceNoneStringWithNone(message):
async def _handle_generate(websocket, message):
global args
# update args parameters which control the model settings
if message.get('autoregressive_model'):
args.autoregressive_model = message['autoregressive_model']
update_autoregressive_model(message['autoregressive_model'])
if message.get('diffusion_model'):
args.diffusion_model = message['diffusion_model']
update_diffusion_model(message['diffusion_model'])
if message.get('tokenizer_json'):
args.tokenizer_json = message['tokenizer_json']
update_tokenizer(message['tokenizer_json'])
if message.get('sample_batch_size'):
global args
args.sample_batch_size = message['sample_batch_size']
message['result'] = generate(**message)