Error when running convert_diffusers_to_flux

[laolongboy]

I want to convert original flux model to bfl format in advance, because convert during training take too long.

TypeError: make_diffusers_to_bfl_map() missing 2 required positional arguments: 'num_double_blocks' and 'num_single_blocks'

https://github.com/kohya-ss/sd-scripts/blob/sd3/tools/convert_diffusers_to_flux.py#L58-L59

https://github.com/kohya-ss/sd-scripts/blob/sd3/library/flux_utils.py#L428-L447

no parameters passed to the func.

[rockerBOO]

If you want to make a PR for that, pass in make_diffusers_to_bfl_map(19, 38) should fix it

[laolongboy]

If you want to make a PR for that, pass in make_diffusers_to_bfl_map(19, 38) should fix it

I tried it and it works! I haven't make any PR before but I will try. Thanks a lot.

[Manni1000]

I think it's strange that this script is hardcoded to use 3 separate safetensors files. I just wanted to convert one Diffusers Flux model that had all the transformer stuff in one safetensor and not 3, and the script failed in many different ways. i have attached a version that worded for me

convert_diffusers_to_flux.py

[kohya-ss]

Because when I created the script, only three-part Diffusers weights were available. If there is only one safetensor format, I will support it.

[rockerBOO]

Yeah the SRPO from tencent they released as a single safetensors file for the 'transformers' part.

https://huggingface.co/tencent/SRPO
https://huggingface.co/rockerBOO/flux.1-dev-SRPO

[rockerBOO]

I used this script to convert it for example.

from library.flux_utils import load_flow_model
from datetime import datetime
from safetensors.torch import save_model

model_path = "/mnt/2t870/models/image/SRPO/flux.1-dev-SRPO-f32.safetensors"
_, model = load_flow_model(model_path, None, "cpu")

print(model)

metadata = {
    "original": "https://huggingface.co/tencent/SRPO",
    "converted_by": "rockerBOO (Dave Lage)",
    "date": datetime.now().isoformat(),
}

save_model(model, "/mnt/2t870/models/image/SRPO/flux.1-dev-SRPO-BFL-f32.safetensors", metadata=metadata)