dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-29 14:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

Support combined voices

Browse source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2025-01-16 20:07:54 +01:00
parent d5ec14daf8
commit eb770fcd92
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
backend/python/kokoro/backend.py
View file

@ -69,8 +69,17 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
if VOICE_NAME is None: if VOICE_NAME is None:
return backend_pb2.Result(success=False, message=f"No voice specified in options") return backend_pb2.Result(success=False, message=f"No voice specified in options")
MODELPATH = request.ModelPath MODELPATH = request.ModelPath
# If voice name contains a plus, split it and load the two models and combine them
if "+" in VOICE_NAME:
voice1, voice2 = VOICE_NAME.split("+")
voice1 = torch.load(f'{MODELPATH}/{voice1}.pt', weights_only=True).to(device)
voice2 = torch.load(f'{MODELPATH}/{voice2}.pt', weights_only=True).to(device)
self.VOICEPACK = torch.mean(torch.stack([voice1, voice2]), dim=0)
else:
self.VOICEPACK = torch.load(f'{MODELPATH}/{VOICE_NAME}.pt', weights_only=True).to(device)
self.VOICE_NAME = VOICE_NAME self.VOICE_NAME = VOICE_NAME
self.VOICEPACK = torch.load(f'{MODELPATH}/{VOICE_NAME}.pt', weights_only=True).to(device)
print(f'Loaded voice: {VOICE_NAME}') print(f'Loaded voice: {VOICE_NAME}')
except Exception as err: except Exception as err:
return backend_pb2.Result(success=False, message=f"Unexpected {err=}, {type(err)=}") return backend_pb2.Result(success=False, message=f"Unexpected {err=}, {type(err)=}")