This commit is contained in:
unknown
2025-12-25 04:58:58 -08:00
parent 263e018792
commit a0191f9b6c
6 changed files with 22 additions and 2 deletions

View File

@@ -234,6 +234,10 @@ class Bark(TTSUtils, TTSRegistry, name='bark'):
error = f"Cannot create {final_sentence_file}"
print(error)
return False
else:
error = f"audio_tensor not valid"
print(error)
return False
else:
error = f"audio_sentence not valid"
print(error)

View File

@@ -203,6 +203,10 @@ class Fairseq(TTSUtils, TTSRegistry, name='fairseq'):
error = f"Cannot create {final_sentence_file}"
print(error)
return False
else:
error = f"audio_tensor not valid"
print(error)
return False
else:
error = f"audio_sentence not valid"
print(error)

View File

@@ -231,6 +231,10 @@ class Tacotron2(TTSUtils, TTSRegistry, name='tacotron'):
error = f"Cannot create {final_sentence_file}"
print(error)
return False
else:
error = f"audio_tensor not valid"
print(error)
return False
else:
error = f"audio_sentence not valid"
print(error)

View File

@@ -216,6 +216,10 @@ class Vits(TTSUtils, TTSRegistry, name='vits'):
error = f"Cannot create {final_sentence_file}"
print(error)
return False
else:
error = f"audio_tensor not valid"
print(error)
return False
else:
error = f"audio_sentence not valid"
print(error)

View File

@@ -146,8 +146,8 @@ class XTTSv2(TTSUtils, TTSRegistry, name='xtts'):
print(error)
return False
audio_tensor = audio_sentence.detach().cpu().unsqueeze(0)
if sentence[-1].isalnum() or sentence[-1] == '':
audio_tensor = trim_audio(audio_tensor.squeeze(), self.params['samplerate'], 0.001, trim_audio_buffer).unsqueeze(0)
if sentence[-1].isalnum() or sentence[-1] == '':
audio_tensor = trim_audio(audio_tensor.squeeze(), self.params['samplerate'], 0.001, trim_audio_buffer).unsqueeze(0)
if audio_tensor is not None and audio_tensor.numel() > 0:
self.audio_segments.append(audio_tensor)
if not re.search(r'\w$', sentence, flags=re.UNICODE) and sentence[-1] != '':

View File

@@ -146,6 +146,10 @@ class YourTTS(TTSUtils, TTSRegistry, name='yourtts'):
error = f"Cannot create {final_sentence_file}"
print(error)
return False
else:
error = f"audio_tensor not valid"
print(error)
return False
else:
error = f"audio_sentence not valid"
print(error)