CalvinBackup/ComfyUI-DeepFuze
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update tts_generation.py

Browse Source
This commit is contained in:
Sam Khoze
2024-06-16 16:24:22 +05:30
committed by GitHub
parent f6fac7114d
commit 2d95749eec
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts_generation.py
+6 -3
View File
@@ -21,10 +21,13 @@ def main():
args = parser.parse_args() args = parser.parse_args()
if args.device == "cuda":
device="cuda"
else:
device="cpu"
# Init TTS # Init TTS
tts = TTS() tts = TTS(model_path=args.model,config_path=os.path.join(args.model,"config.json")).to(device)
tts.load_tts_model_by_path(model_path=args.model,config_path=os.path.join(args.model,"config.json"))
# tts.to(args.device)
# Run TTS and save to file # Run TTS and save to file
tts.tts_to_file(text=args.text, speaker_wav=args.speaker_wav, language=args.language, file_path=args.output_file) tts.tts_to_file(text=args.text, speaker_wav=args.speaker_wav, language=args.language, file_path=args.output_file)