KNNet does not support FP16 (#36)

chaiNNer-org · Nov 22, 2023 · 80ce019 · 80ce019
1 parent a6473e9
commit 80ce019
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spandrel/architectures/KBNet/__init__.py b/src/spandrel/architectures/KBNet/__init__.py
@@ -61,7 +61,7 @@ def load_l(state_dict: StateDict) -> RestorationModelDescriptor[KBNet_l]:
         state_dict,
         architecture="KBCNN",
         tags=["L"],
-        supports_half=True,
+        supports_half=False,
         supports_bfloat16=True,
         input_channels=in_nc,
         output_channels=out_nc,
@@ -121,7 +121,7 @@ def load_s(state_dict: StateDict) -> RestorationModelDescriptor[KBNet_s]:
         state_dict,
         architecture="KBCNN",
         tags=["S"],
-        supports_half=True,
+        supports_half=False,
         supports_bfloat16=True,
         input_channels=img_channel,
         output_channels=img_channel,