diff --git a/dumas_h3_longvideos.py b/dumas_h3_longvideos.py index 3e636c0..4b058d5 100644 --- a/dumas_h3_longvideos.py +++ b/dumas_h3_longvideos.py @@ -6615,11 +6615,12 @@ class H3LongVideos: detail += f"/{latent_upscale_param.get('method', 'bilinear')}" denoise_value = latent_upscale_param.get("denoise", latent_upscale_param.get("refine_denoise", 0.2)) denoise = 0.2 if denoise_value is None else float(denoise_value) + refine_sampler = latent_upscale_param.get("sampler_name", "euler_ancestral") + refine_scheduler = latent_upscale_param.get("scheduler", "simple") latent_upscale_note = ( f" latent upscale: target {target_w}x{target_h}px{detail}; " f"{int(latent_upscale_param.get('steps', 2) or 2)}-step refinement " - f"{latent_upscale_param.get('sampler_name', sn)}/{latent_upscale_param.get('scheduler', sch)} " - f"denoise {denoise:.2f}" + f"{refine_sampler}/{refine_scheduler} denoise {denoise:.2f}" ) paras = split_paragraphs(prompt, "##") diff --git a/tests/test_dumas_h3_longvideos.py b/tests/test_dumas_h3_longvideos.py index d3f33d5..8f598f9 100644 --- a/tests/test_dumas_h3_longvideos.py +++ b/tests/test_dumas_h3_longvideos.py @@ -966,12 +966,23 @@ class DumasH3LongVideosHelperTests(unittest.TestCase): plan_only=True, ref_1={"image": "live-1"}, ref_3={"image": "live-3"}, + latent_upscale_param={ + "mode": "interp", + "method": "bilinear", + "sampler_name": "euler_ancestral", + "scheduler": "simple", + "steps": 2, + "denoise": 0.2, + "megapixels": 1.5, + }, ) self.assertEqual(calls["refs"][0]["image"], "live-1") self.assertIsNone(calls["refs"][1]) self.assertEqual(calls["refs"][2]["image"], "live-3") self.assertEqual(result[2].count("ref2va: 2 reference image(s)"), 1) + self.assertIn("latent upscale:", result[2]) + self.assertIn("euler_ancestral/simple", result[2]) finally: self.module.parse_resolution = original_parse_resolution self.module._connected_refs = original_connected_refs