fix(api): keep kernel slices within bounds

2023-06-16 20:49:43 -05:00 · 2023-06-16 20:49:43 -05:00 · 5713957026
parent c1a4484b75
commit 5713957026
1 changed files with 2 additions and 2 deletions
--- 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),
    )