mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
mrq 2023-03-14 16:23:29 +00:00
parent 65fe304267
commit 9d2c7fb942
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils.py
View File

@ -1048,6 +1048,7 @@ def stop_training():
training_state.killed = True training_state.killed = True
children = [] children = []
if args.tts_backend == "tortoise":
# wrapped in a try/catch in case for some reason this fails outside of Linux # wrapped in a try/catch in case for some reason this fails outside of Linux
try: try:
children = [p.info for p in psutil.process_iter(attrs=['pid', 'name', 'cmdline']) if './src/train.py' in p.info['cmdline']] children = [p.info for p in psutil.process_iter(attrs=['pid', 'name', 'cmdline']) if './src/train.py' in p.info['cmdline']]
@ -1057,6 +1058,9 @@ def stop_training():
training_state.process.stdout.close() training_state.process.stdout.close()
training_state.process.terminate() training_state.process.terminate()
training_state.process.kill() training_state.process.kill()
elif args.tts_backend == "vall-e":
print(training_state.process.communicate(input='quit')[0])
return_code = training_state.process.wait() return_code = training_state.process.wait()
for p in children: for p in children: