Update vocoder_preprocess.py

@Tomcattwo changed line 47 per recommendation by @blue-fish in issue CorentinJ#833; allows --cpu argument to be properly recognized
Tomcattwo · Sep 1, 2021 · 34b13c2 · 34b13c2
1 parent ac46da5
commit 34b13c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vocoder_preprocess.py b/vocoder_preprocess.py
@@ -43,7 +43,8 @@ class MyFormatter(argparse.ArgumentDefaultsHelpFormatter, argparse.RawDescriptio
 
     if args.cpu:
         # Hide GPUs from Pytorch to force CPU processing
-        os.environ["CUDA_VISIBLE_DEVICES"] = ""
+        #@tomcattwo changed line 47 per recommendation by @blue-fish in issue #833; allows --cpu argument to be properly recognized
+        os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
     # Verify webrtcvad is available
     if not args.no_trim: