Back off latent upscale tile size on OOM

2026-09-03 19:43:15 +00:00
parent 3b63d33ec7
commit 836a6eba33
2 changed files with 56 additions and 0 deletions
@@ -1160,6 +1160,20 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
        self.assertIs(tagged, exc)
        self.assertEqual(getattr(tagged, "_h3_stage", ""), "latent_upscale")

+    def test_shrink_model_tile_param_reduces_tile_size(self):
+        latent = importlib.import_module("dumas_h3_latent_upscale")
+        smaller = latent._shrink_model_tile_param({
+            "tile_size_mode": "specific_size",
+            "tile_width": 512,
+            "tile_height": 512,
+            "overlap": 64,
+            "fade_width": 32,
+            "fade_height": 32,
+        })
+        self.assertIsNotNone(smaller)
+        self.assertLess(smaller["tile_width"], 512)
+        self.assertLess(smaller["tile_height"], 512)
+
    def test_compose_persistent_does_not_expand_ambiguous_plural_to_full_cast(self):
        active = self.module.parse_wardrobe(
            "Maya = she, red jacket\n"