bugfix
This commit is contained in:
parent
33c535bec4
commit
2b4d7594dc
@ -14,6 +14,7 @@ from cached_path import cached_path
|
|||||||
from appPublic.textsplit import split_text_with_dialog_preserved
|
from appPublic.textsplit import split_text_with_dialog_preserved
|
||||||
from appPublic.uniqueID import getID
|
from appPublic.uniqueID import getID
|
||||||
from ahserver.serverenv import get_serverenv
|
from ahserver.serverenv import get_serverenv
|
||||||
|
from filetxt.loader import fileloader
|
||||||
import pycld2 as cld
|
import pycld2 as cld
|
||||||
import cn2an
|
import cn2an
|
||||||
|
|
||||||
@ -199,14 +200,19 @@ class F5TTS:
|
|||||||
debug(f'{gen_text=} inference error\n{format_exc()}')
|
debug(f'{gen_text=} inference error\n{format_exc()}')
|
||||||
|
|
||||||
async def inference_stream(self, prompt, speaker, speed_factor=1.0):
|
async def inference_stream(self, prompt, speaker, speed_factor=1.0):
|
||||||
|
total_duration = 0
|
||||||
async for d in self._inference_stream(prompt, speaker, speed_factor=speed_factor):
|
async for d in self._inference_stream(prompt, speaker, speed_factor=speed_factor):
|
||||||
|
sampels = d['audio'].shape[0]
|
||||||
|
duration = samples / d['sample_rate']
|
||||||
audio_b64=audio_ndarray_to_base64(d['audio'], d['sample_rate'])
|
audio_b64=audio_ndarray_to_base64(d['audio'], d['sample_rate'])
|
||||||
d['audio'] = audio_b64
|
d['audio'] = audio_b64
|
||||||
|
d['duration'] = duration
|
||||||
d['done'] = False
|
d['done'] = False
|
||||||
txt = json.dumps(d, ensure_ascii=False)
|
txt = json.dumps(d, ensure_ascii=False)
|
||||||
yield txt + '\n'
|
yield txt + '\n'
|
||||||
d = {
|
d = {
|
||||||
'done': True
|
'done': True,
|
||||||
|
'duration': total_duration
|
||||||
}
|
}
|
||||||
txt = json.dumps(d, ensure_ascii=False)
|
txt = json.dumps(d, ensure_ascii=False)
|
||||||
yield txt + '\n'
|
yield txt + '\n'
|
||||||
@ -329,6 +335,7 @@ def init():
|
|||||||
g.test1 = awaitify(test1)
|
g.test1 = awaitify(test1)
|
||||||
g.add_voice = f5.add_voice
|
g.add_voice = f5.add_voice
|
||||||
g.UiError = UiError
|
g.UiError = UiError
|
||||||
|
g.filelaoder = fileloader
|
||||||
g.UiMessage = UiMessage
|
g.UiMessage = UiMessage
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
@ -6,4 +6,5 @@ cn2an
|
|||||||
apppublic
|
apppublic
|
||||||
sqlor
|
sqlor
|
||||||
ahserver
|
ahserver
|
||||||
|
file2text
|
||||||
# git+https://github.com/SWivid/F5-TTS
|
# git+https://github.com/SWivid/F5-TTS
|
||||||
|
Loading…
Reference in New Issue
Block a user