Double GPU performance

roop/core.py

@@ -1,11 +1,12 @@
 #!/usr/bin/env python3
 
 import os
-# reducing num threads doubles performance of gpu-mode
-os.environ['OMP_NUM_THREADS'] = '1'
+import sys
+# single thread doubles performance of gpu-mode - needs to be set before torch import
+if any(arg.startswith('--gpu-vendor=') for arg in sys.argv):
+    os.environ['OMP_NUM_THREADS'] = '1'
 import platform
 import signal
-import sys
 import shutil
 import glob
 import argparse