Cool_Tools/ultimatevocalremovergui
1
0
Fork 0
mirror of https://github.com/Anjok07/ultimatevocalremovergui.git synced 2024-11-12 10:00:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update inference_v5_ensemble.py

Browse Source
This commit is contained in:
Anjok07 2022-04-17 15:39:50 -05:00 committed by GitHub
parent ae03754942
commit ef58d60482
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
inference_v5_ensemble.py
View File

@ -501,6 +501,14 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
_, bin_image = cv2.imencode('.jpg', image)
bin_image.tofile(f)
text_widget.write(base_text + 'Clearing CUDA Cache... ')
torch.cuda.empty_cache()
time.sleep(3)
text_widget.write('Done!\n')
text_widget.write(base_text + 'Completed Seperation!\n\n')
# Emsembling Outputs