Script to merge Flux diffuser into 1 safetensor file

[pivtienduc]

Hi everyone

I found Flux model which is split into 3 parts. So far I can't find any script to merge splitted parts of Flux model into one

Can anyone help me to do it

Thank in advance

[a-r-r-o-w]

You can load the model using from_pretrained and save it again, but by specifying max_shard_size="30GB" to get a single large file. The docs for loading/saving pipelines contains more info about from_pretrained and save_pretrained.