Unload interum weights when doing multi lora fuse

This commit is contained in:
Jaret Burkett committed 2024-08-18 09:35:10 -06:00
1 parent 13a965a26c
commit c45887192a
1 file changed
+2
+2
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
)