Fix latent upscale summary scope leak
This commit is contained in:
@@ -6615,11 +6615,12 @@ class H3LongVideos:
|
|||||||
detail += f"/{latent_upscale_param.get('method', 'bilinear')}"
|
detail += f"/{latent_upscale_param.get('method', 'bilinear')}"
|
||||||
denoise_value = latent_upscale_param.get("denoise", latent_upscale_param.get("refine_denoise", 0.2))
|
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)
|
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 = (
|
latent_upscale_note = (
|
||||||
f" latent upscale: target {target_w}x{target_h}px{detail}; "
|
f" latent upscale: target {target_w}x{target_h}px{detail}; "
|
||||||
f"{int(latent_upscale_param.get('steps', 2) or 2)}-step refinement "
|
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"{refine_sampler}/{refine_scheduler} denoise {denoise:.2f}"
|
||||||
f"denoise {denoise:.2f}"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
paras = split_paragraphs(prompt, "##")
|
paras = split_paragraphs(prompt, "##")
|
||||||
|
|||||||
@@ -966,12 +966,23 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
|
|||||||
plan_only=True,
|
plan_only=True,
|
||||||
ref_1={"image": "live-1"},
|
ref_1={"image": "live-1"},
|
||||||
ref_3={"image": "live-3"},
|
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.assertEqual(calls["refs"][0]["image"], "live-1")
|
||||||
self.assertIsNone(calls["refs"][1])
|
self.assertIsNone(calls["refs"][1])
|
||||||
self.assertEqual(calls["refs"][2]["image"], "live-3")
|
self.assertEqual(calls["refs"][2]["image"], "live-3")
|
||||||
self.assertEqual(result[2].count("ref2va: 2 reference image(s)"), 1)
|
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:
|
finally:
|
||||||
self.module.parse_resolution = original_parse_resolution
|
self.module.parse_resolution = original_parse_resolution
|
||||||
self.module._connected_refs = original_connected_refs
|
self.module._connected_refs = original_connected_refs
|
||||||
|
|||||||
Reference in New Issue
Block a user