fix(api): correct name for kwargs

ssube · Jan 22, 2023 · 9bff64c · 9bff64c
1 parent 986112e
commit 9bff64c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py
@@ -75,10 +75,10 @@ def eval(self) -> None:
     def half(self):
         return self
 
-    def load_state_dict(self, _net, _strict=True) -> None:
+    def load_state_dict(self, net, strict=True) -> None:
         pass
 
-    def to(self, _device):
+    def to(self, device):
         return self