Allow loading StoryMem -LoRAs
https://huggingface.co/Kevin-thu/StoryMem
This commit is contained in:
@@ -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__'):
|
||||||
|
|||||||
Reference in New Issue
Block a user