As mentioned in this issue by @naveenkumaryp we need to explicitly define prompt as a key: https://huggingface.co/black-forest-labs/FLUX.2-klein-4B/discussions/3

Other Klein models also have this problem.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment