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

@@ -60,7 +60,6 @@ class SPSRNet(nn.Module):
self.out_nc: int = self.state["f_HR_conv1.0.bias"].shape[0]
self.scale = self.get_scale(4)
print(self.scale)
self.num_filters: int = self.state["model.0.weight"].shape[0]
self.supports_fp16 = True