how to load and use dit.bin obtained by "train_flux_deepspeed.py" to generate images?

[yinguoweiOvO]

When I finetuned flux.dev model on my custom dataset using train_flux_deepspeed.py(not LoRA !!!), this code saved the checkpoint file (dit.bin).

I want to know how to generate images with the checkpoint dit.bin file.

[godwenbin]

@yinguoweiOvO have you known how to load the .bin file to generate pics?

[yinguoweiOvO]

@yinguoweiOvO have you known how to load the .bin file to generate pics?

Sorry, I didn't solve it. And you can train flux using the code from diffusers or sd_scripts, maybe they haven't this problem.