Update upscale model code to latest Chainner model code.

Don't add SRFormer because the code license is incompatible with the GPL.

Remove MAT because it's unused and the license is incompatible with GPL.
This commit is contained in:
comfyanonymous
2023-09-02 22:25:12 -04:00
parent 62efc78a4b
commit 766c7b3815
12 changed files with 2084 additions and 2511 deletions

View File

@@ -972,6 +972,7 @@ class SwinIR(nn.Module):
self.upsampler = upsampler
self.img_size = img_size
self.img_range = img_range
self.resi_connection = resi_connection
self.supports_fp16 = False # Too much weirdness to support this at the moment
self.supports_bfp16 = True