fix(api): keep kernel slices within bounds

ssube · Jun 17, 2023 · 5713957 · 5713957
1 parent c1a4484
commit 5713957
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/convert/diffusion/lora.py b/api/onnx_web/convert/diffusion/lora.py
@@ -79,8 +79,8 @@ def fix_node_name(key: str):
 
 def kernel_slice(x: int, y: int, shape: Tuple[int, int, int, int]) -> Tuple[int, int]:
     return (
-        max(x, shape[2]),
-        max(y, shape[3]),
+        min(x, shape[2] - 1),
+        min(y, shape[3] - 1),
     )