new xtts module and modified requirements.txt

This commit is contained in:
Tasha Upchurch
2024-10-20 15:55:20 -06:00
parent 8f068ea306
commit 06d4f50608
2 changed files with 8 additions and 6 deletions
+3 -1
View File
@@ -438,6 +438,7 @@ class XttsNode:
"--folder", kwargs.get('output_path', "."),
"--lang", kwargs.get('lang', 'en'),
"--speaker", kwargs.get('speaker_wav', ''),
"--device", kwargs.get('device', 'cuda:0'),
]
# write the command to "temp_command.txt" in the same folder as this file is in
with open(os.path.join(this_file_path, "temp_command.txt"), "w") as f:
@@ -446,7 +447,8 @@ class XttsNode:
# execute the command in its own full porcess and recieve the output
try:
# Capture both stdout and stderr
output = subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT)
#output = subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT)
output = subprocess.check_output(command, stderr=subprocess.STDOUT)
generated = output.decode("utf-8")
if "Generated files: (" in generated:
+5 -5
View File
@@ -128,7 +128,7 @@ def main():
parser.add_argument('--text', help="Text to synthesize.")
parser.add_argument('--file', help="File containing text to synthesize (one text per line).")
parser.add_argument('--lang', default='en', help="Language for the TTS.")
parser.add_argument('--output', help="Path to save the output TTS file.")
parser.add_argument('--folder', help="Path to save the output TTS file.")
parser.add_argument('--device', default='cuda:0', help="Device to run the model on. Default: cuda:0")
parser.add_argument('--test', action='store_true', help="Run a test TTS using the test.speech file.")
@@ -145,9 +145,9 @@ def main():
raise FileNotFoundError(f"The test file '{default_test_file_path}' does not exist. Please create it.")
args.file = default_test_file_path
args.output = os.path.join(script_dir, 'test_output.wav')
args.folder = os.path.join(script_dir, 'test_output.wav')
print(f"Test file: {args.file}")
print(f"Output will be saved to: {args.output}")
print(f"Output will be saved to: {args.folder}")
# Handle the --file argument
if args.file:
@@ -172,7 +172,7 @@ def main():
device=args.device
)
output_file = f"{args.output}_{i}.wav" if args.output else f'./output_{i}.wav'
output_file = f"{args.folder}_{i}.wav" if args.folder else f'./output_{i}.wav'
try:
os.rename(output_path, output_file)
except Exception as e:
@@ -192,7 +192,7 @@ def main():
device=args.device
)
output_file = args.output if args.output else './output.wav'
output_file = args.folder if args.folder else './output.wav'
try:
os.rename(output_path, output_file)
except Exception as e: