edicam
envy-ai
/ hidream_diffusers_merge_ai-toolkit_lora.py
Created
May 14, 2025 16:14
Merge ai-toolkit lora into diffusers HiDream model.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| import os | |
| import argparse | |
| import torch | |
| from safetensors.torch import load_file, save_file | |
| # NOTE: When done, move the newly created transformer into the transformer/ subdirectory of your diffusers model, and | |
| # remove the old .safetensors files and the .index.json file (which is no longer needed because it's not sharded) | |
| def manual_merge( |