Supplement the input parameters to correctly convert the flux model to BFL format; fixes #1996

This commit is contained in:
laolongboy
2025-03-24 23:33:25 +08:00
parent 8ebe858f89
commit e64dc05c2a

View File

@@ -56,7 +56,7 @@ def convert(args):
save_dtype = str_to_dtype(args.save_precision) if args.save_precision is not None else None
# make reverse map from diffusers map
diffusers_to_bfl_map = flux_utils.make_diffusers_to_bfl_map()
diffusers_to_bfl_map = flux_utils.make_diffusers_to_bfl_map(19, 38)
# iterate over three safetensors files to reduce memory usage
flux_sd = {}