It looks like most other pipelines such as Flux 2 use the following callback_kwargs pattern instead of a dict comprehension:
|
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) |
If I change the Ideogram 4 pipeline to use the above pattern, the test passes locally for me. So I think we should probably fix this in src/diffusers/pipelines/ideogram4/pipeline_ideogram4.py, then unskip this test (as well as test_callback_cfg below, which also passes after the change).
Originally posted by @dg845 in #14599 (comment)
It looks like most other pipelines such as Flux 2 use the following
callback_kwargspattern instead of a dict comprehension:diffusers/src/diffusers/pipelines/flux2/pipeline_flux2.py
Lines 996 to 999 in 82f175e
If I change the Ideogram 4 pipeline to use the above pattern, the test passes locally for me. So I think we should probably fix this in
src/diffusers/pipelines/ideogram4/pipeline_ideogram4.py, then unskip this test (as well astest_callback_cfgbelow, which also passes after the change).Originally posted by @dg845 in #14599 (comment)