diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py
index 0e9faa15..22dd5fc5 100644
--- a/api/onnx_web/upscale.py
+++ b/api/onnx_web/upscale.py
@@ -48,7 +48,7 @@ class ONNXNet():
     def half(self):
         return self
 
-    def load_state_dict(self) -> None:
+    def load_state_dict(self, net, strict=True) -> None:
         pass
 
     def to(self, device):