added DirectX12-compatible cards support through onnxruntime-directml

main.py

@@ -1,5 +1,9 @@
+import argparse
 import os
 import platform
+from pathlib import Path
+
+from xlib import appargs as lib_appargs
 
 # onnxruntime==1.8.0 requires CUDA_PATH_V11_2, but 1.8.1 don't
 # keep the code if they return that behaviour
@@ -10,9 +14,6 @@ import platform
 #         # set environ for onnxruntime
 #         # os.environ['CUDA_PATH_V11_2'] = os.environ['CUDA_PATH']
 
-import argparse
-from pathlib import Path
-
 def main():
     parser = argparse.ArgumentParser()
     subparsers = parser.add_subparsers()
@@ -22,29 +23,31 @@ def main():
 
     def run_DeepFaceLive(args):
         userdata_path = Path(args.userdata_dir)
+        lib_appargs.set_arg_bool('NO_CUDA', args.no_cuda)
+
         print('Running DeepFaceLive.')
         from apps.DeepFaceLive.DeepFaceLiveApp import DeepFaceLiveApp
         DeepFaceLiveApp(userdata_path=userdata_path).run()
 
     p = run_subparsers.add_parser('DeepFaceLive')
     p.add_argument('--userdata-dir', default=None, action=fixPathAction, help="Workspace directory.")
+    p.add_argument('--no-cuda', action="store_true", default=False, help="Disable CUDA.")
     p.set_defaults(func=run_DeepFaceLive)
 
-
     dev_parser = subparsers.add_parser("dev")
     dev_subparsers = dev_parser.add_subparsers()
-    
+
     def run_split_large_files(args):
         from scripts import dev
         dev.split_large_files()
-    
+
     p = dev_subparsers.add_parser('split_large_files')
     p.set_defaults(func=run_split_large_files)
-    
+
     def run_merge_large_files(args):
         from scripts import dev
         dev.merge_large_files(delete_parts=args.delete_parts)
-    
+
     p = dev_subparsers.add_parser('merge_large_files')
     p.add_argument('--delete-parts', action="store_true", default=False)
     p.set_defaults(func=run_merge_large_files)
@@ -63,6 +66,6 @@ class fixPathAction(argparse.Action):
 
 if __name__ == '__main__':
     main()
-    
+
 # import code
 # code.interact(local=dict(globals(), **locals()))