Remove BSRGAN from --use-cpu, add SwinIR

This commit is contained in:
brkirch
2022-10-24 23:04:50 -04:00
committed by AUTOMATIC1111
parent f53ca51638
commit 4c24347e45
3 changed files with 10 additions and 10 deletions

View File

@@ -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