From fe657468bf3ae6011690eac1de07a1842dc9e82f Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Mon, 16 Jan 2023 10:59:47 -0600
Subject: [PATCH] fix(api): add missing params to load stub

---
 api/onnx_web/upscale.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):