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')}"
|
||||
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, "##")
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user