Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2025-02-21 21:00:15 +01:00
Code Releases Wiki Activity

fix extract feature in MPS device

Browse Source
This commit is contained in:
源文雨 2023-06-24 16:21:31 +08:00
parent 359ba54321
commit 46c0e9b2fe
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extract_feature_print.py
View File

@ -1,5 +1,7 @@
import os, sys, traceback import os, sys, traceback
os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
# device=sys.argv[1] # device=sys.argv[1]
n_part = int(sys.argv[2]) n_part = int(sys.argv[2])
i_part = int(sys.argv[3]) i_part = int(sys.argv[3])