fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-07 10:10:23 +08:00
Code Activity

Fix memory doesn't unload on mps device (#2035)

Browse Source 
* Solving the cache not being cleared in mps

* Fix protect not to be dict
This commit is contained in:
Ftps
2024-05-05 15:22:37 +09:00
committed by GitHub
parent fb16ba36b9
commit d255818097
3 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
infer/modules/uvr5/modules.py
View File

@@ -105,4 +105,7 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
if torch.cuda.is_available():
torch.cuda.empty_cache()
logger.info("Executed torch.cuda.empty_cache()")
elif torch.backends.mps.is_available():
torch.mps.empty_cache()
logger.info("Executed torch.mps.empty_cache()")
yield "\n".join(infos)