Update custom-hires-fix-mod-for-automatic1111-2.9/scripts/custom_hires_fix.py

custom-hires-fix-mod-for-automatic1111-2.9/scripts/custom_hires_fix.py

@@ -135,6 +135,17 @@ config_path = (Path(__file__).parent.resolve() / "../config.yaml").resolve()
 
 
 class CustomHiresFix(scripts.Script):
+    def _get_free_vram_bytes(self):
+        """Return free VRAM bytes if available; else None."""
+        try:
+            import torch
+            if torch.cuda.is_available():
+                free, total = torch.cuda.mem_get_info()
+                return int(free)
+        except Exception:
+            pass
+        return None
+
     def _interp(self, *args, _default_mode: str = "bicubic", **kwargs):
         """Config-driven interpolate with runtime fallbacks."""
         import torch.nn.functional as F
@@ -142,8 +153,16 @@ class CustomHiresFix(scripts.Script):
             label = str(self.config.get("latent_resample_mode", _default_mode))
             mode, antialias = self._parse_interpolate_mode(label)
             kwargs["mode"] = mode
-            kwargs.setdefault("align_corners", False)
-            kwargs["antialias"] = antialias
+            # Set align_corners only for linear/bilinear/bicubic/trilinear families
+            if mode in {"linear", "bilinear", "bicubic", "trilinear"}:
+                kwargs.setdefault("align_corners", False)
+            else:
+                kwargs.pop("align_corners", None)
+            # antialias valid only for linear/bilinear/bicubic/trilinear
+            if mode in {"linear", "bilinear", "bicubic", "trilinear"}:
+                kwargs["antialias"] = antialias
+            else:
+                kwargs.pop("antialias", None)
         try:
             return F.interpolate(*args, **kwargs)
         except Exception as e: