Remove BSRGAN from --use-cpu, add SwinIR

2022-10-24 23:04:50 -04:00
parent df0a1f8381
commit 355acddcd7
3 changed files with 10 additions and 10 deletions
@@ -45,7 +45,7 @@ def enable_tf32():

 errors.run(enable_tf32, "Enabling TF32")

-device = device_interrogate = device_gfpgan = device_bsrgan = device_esrgan = device_scunet = device_codeformer = None
+device = device_interrogate = device_gfpgan = device_swinir = device_esrgan = device_scunet = device_codeformer = None
 dtype = torch.float16
 dtype_vae = torch.float16