Allow loading StoryMem -LoRAs

https://huggingface.co/Kevin-thu/StoryMem
This commit is contained in:
kijai
2025-12-23 22:26:37 +02:00
parent f28e7da442
commit c5d3fb450c
+1
View File
@@ -132,6 +132,7 @@ def standardize_lora_key_format(lora_sd):
k = k.replace('vace_blocks.', 'diffusion_model.vace_blocks.') k = k.replace('vace_blocks.', 'diffusion_model.vace_blocks.')
k = k.replace('.default.', '.') k = k.replace('.default.', '.')
k = k.replace('.diff_m', '.modulation.diff') k = k.replace('.diff_m', '.modulation.diff')
k = k.replace('base_model.model.', 'diffusion_model.')
# Fun LoRA format # Fun LoRA format
if k.startswith('lora_unet__'): if k.startswith('lora_unet__'):