ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unload interum weights when doing multi lora fuse

Browse Source
This commit is contained in:
Jaret Burkett
2024-08-18 09:35:10 -06:00
parent 13a965a26c
commit c45887192a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/stable_diffusion_model.py
View File

@@ -557,6 +557,7 @@ class StableDiffusion:
)
pipe.load_lora_weights(double_transformer_lora, adapter_name=f"lora1_double")
pipe.fuse_lora()
pipe.unload_lora_weights()
transformer.transformer_blocks = transformer.transformer_blocks.to(
'cpu', dtype=dtype
)
@@ -567,6 +568,7 @@ class StableDiffusion:
)
pipe.load_lora_weights(single_transformer_lora, adapter_name=f"lora1_single")
pipe.fuse_lora()
pipe.unload_lora_weights()
transformer.single_transformer_blocks = transformer.single_transformer_blocks.to(
'cpu', dtype=dtype
)