Skip to content

It looks like most other pipelines such as Flux 2 use the following callback_kwargs pattern instead of a dict comprehension: #14617

Description

@sayakpaul

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)

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions