| from safetensors.torch import load_file, save_file | |
| import torch | |
| # --- Input and output file paths --- | |
| src = "hlpvWkBrqzopjecZbuVIA_pytorch_lora_weights.safetensors" | |
| dst = "hlpvWkBrqzopjecZbuVIA_pytorch_lora_weights_sdxl.safetensors" | |
| # --- Load original LoRA weights --- | |
| weights = load_file(src) | |
| new_weights = {} | |
| # --- Simple rename mapping for SDXL --- | |
| for k, v in weights.items(): | |
| k2 = k.replace("lora_unet_", "unet.") \ | |
| .replace("lora_te_", "text_encoder.") \ | |
| .replace("lora_te2_", "text_encoder_2.") | |
| new_weights[k2] = v.clone() | |
| # --- Save as SDXL-compatible LoRA --- | |
| save_file(new_weights, dst) | |
| print(f"✅ Converted and saved as {dst}") | |