bruh moment 9

This commit is contained in:
Somdev Sangwan 2023-05-30 07:01:02 +05:30 committed by GitHub
parent 25444da59d
commit 251173ffa9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

2
run.py
View File

@ -42,7 +42,7 @@ parser.add_argument('--keep-frames', help='keep frames directory', dest='keep_fr
for name, value in vars(parser.parse_args()).items():
args[name] = value
if torch.cuda.is_available() not args['gpu']:
if not torch.cuda.is_available() and args['gpu']:
quit("You are using --gpu flag but CUDA isn't available on your system.")
sep = "/"