Stop retrying latent upscale OOMs as sampling

2026-09-03 18:46:56 +00:00
parent 3ab6342ce5
commit c62921c3e8
2 changed files with 41 additions and 20 deletions
@@ -1154,6 +1154,12 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
        self.assertEqual(self.module._latent_upscale_mode({"mode": "model"}), "model")
        self.assertEqual(self.module._latent_upscale_mode({}), "off")

+    def test_tag_oom_stage_marks_oom_exceptions(self):
+        exc = RuntimeError("CUDA out of memory")
+        tagged = self.module._tag_oom_stage(exc, "latent_upscale")
+        self.assertIs(tagged, exc)
+        self.assertEqual(getattr(tagged, "_h3_stage", ""), "latent_upscale")
+
    def test_compose_persistent_does_not_expand_ambiguous_plural_to_full_cast(self):
        active = self.module.parse_wardrobe(
            "Maya = she, red jacket\n"