simplify xfrmers options: --xformers to enable and that's it

2022-10-08 17:02:18 +03:00
parent 7ff1170a2e
commit dc1117233e
4 changed files with 16 additions and 10 deletions
@@ -32,7 +32,7 @@ def extract_arg(args, name):


 args, skip_torch_cuda_test = extract_arg(args, '--skip-torch-cuda-test')
-args, xformers = extract_arg(args, '--xformers')
+xformers = '--xformers' in args


 def repo_dir(name):