Add BSRGAN to --add-cpu

This commit is contained in:
brkirch
2022-10-04 05:18:17 -04:00
parent 2adb249740
commit 27ddc24fde
3 changed files with 7 additions and 7 deletions

View File

@@ -32,7 +32,7 @@ def enable_tf32():
errors.run(enable_tf32, "Enabling TF32")
device = device_gfpgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device()
device = device_gfpgan = device_bsrgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device()
dtype = torch.float16
def randn(seed, shape):