Merge pull request Fannovel16#76 from Visionatrix/fix/mps-border-unsupported

MPS: Unsupported Border padding mode

Author: Fannovel16

vfi_models/rife/rife_arch.py

@@ -56,11 +56,16 @@ def warp(tenInput, tenFlow):
     if tenInput.type() == "torch.cuda.HalfTensor":
         g = g.half()
 
+    padding_mode = "border"
+    if device.type == "mps":
+        # https://github.com/pytorch/pytorch/issues/125098
+        padding_mode = "zeros"
+        g = g.clamp(-1, 1)
     return torch.nn.functional.grid_sample(
         input=tenInput,
         grid=g,
         mode="bilinear",
-        padding_mode="border",
+        padding_mode=padding_mode,
         align_corners=True,
     )