From acc8d057d6a45d827d9e24491a16d3754e0f48a7 Mon Sep 17 00:00:00 2001 From: songh <1020258195@qq.com> Date: Wed, 15 Apr 2026 16:53:16 +0800 Subject: [PATCH] fix(ernie-image): avoid locals() comprehension scope issue in callback kwargs --- src/diffusers/pipelines/ernie_image/pipeline_ernie_image.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/ernie_image/pipeline_ernie_image.py b/src/diffusers/pipelines/ernie_image/pipeline_ernie_image.py index 9fbeee3395ec..64fb2d050019 100644 --- a/src/diffusers/pipelines/ernie_image/pipeline_ernie_image.py +++ b/src/diffusers/pipelines/ernie_image/pipeline_ernie_image.py @@ -352,7 +352,9 @@ def __call__( # Callback if callback_on_step_end is not None: - callback_kwargs = {k: locals()[k] for k in callback_on_step_end_tensor_inputs} + callback_kwargs = {} + for k in callback_on_step_end_tensor_inputs: + callback_kwargs[k] = locals()[k] callback_outputs = callback_on_step_end(self, i, t, callback_kwargs) latents = callback_outputs.pop("latents", latents)