Expose latent upscale sampler controls

This commit is contained in:
2026-09-03 13:05:32 +00:00
parent c1d937e0e2
commit 5c12cd18a3
5 changed files with 126 additions and 52 deletions
+18 -7
View File
@@ -306,18 +306,21 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
latent_upscale_param={
"mode": "model",
"model_name": "upscale.safetensors",
"width": 256,
"height": 128,
"device": "cpu",
"precision": "fp16",
"refine_denoise": 0.4,
"sampler_name": "euler_ancestral",
"scheduler": "simple",
"steps": 2,
"denoise": 0.4,
"megapixels": 1.5,
},
)
self.assertEqual(len(calls), 2)
self.assertIsNot(calls[1][0][8], first_out)
self.assertEqual(calls[1][0][8]["samples"].unbind()[0].name, "upv")
self.assertEqual(calls[1][0][4], "res_multistep")
self.assertEqual(calls[1][0][2], 2)
self.assertEqual(calls[1][0][4], "euler_ancestral")
self.assertEqual(calls[1][0][5], "simple")
self.assertAlmostEqual(calls[1][1]["denoise"], 0.4)
self.assertEqual(result[1], first_out)
@@ -424,11 +427,13 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
latent_upscale_param={
"mode": "model",
"model_name": "upscale.safetensors",
"width": 256,
"height": 128,
"device": "cpu",
"precision": "fp16",
"refine_denoise": 0.4,
"sampler_name": "euler_ancestral",
"scheduler": "simple",
"steps": 2,
"denoise": 0.4,
"megapixels": 1.5,
},
)
@@ -1096,6 +1101,7 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
def test_latent_upscale_params_node_is_exposed(self):
latent = importlib.import_module("dumas_h3_latent_upscale")
required = latent.H3LatentUpscaleParams.INPUT_TYPES()["required"]
self.assertEqual(
latent.NODE_CLASS_MAPPINGS,
@@ -1105,6 +1111,11 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
latent.NODE_DISPLAY_NAME_MAPPINGS,
{"DumasH3LatentUpscaleParams": "Dumas H3 Latent Upscale Params"},
)
self.assertEqual(required["sampler_name"][1]["default"], "euler_ancestral")
self.assertEqual(required["scheduler"][1]["default"], "simple")
self.assertEqual(required["steps"][1]["default"], 2)
self.assertEqual(required["denoise"][1]["default"], 0.2)
self.assertEqual(required["megapixels"][1]["default"], 1.0)
def test_compose_persistent_does_not_expand_ambiguous_plural_to_full_cast(self):
active = self.module.parse_wardrobe(