Skip to content

GitLab

Commit 4a3e5748 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent c2a28c34
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/dreambooth/train_dreambooth_lora.py
View file @ 4a3e5748
...@@ -871,7 +871,7 @@ def main(args): ...@@ -871,7 +871,7 @@ def main(args):
lora_attn_processor_class = ( lora_attn_processor_class = (
LoRAAttnProcessor2_0 if hasattr(F, "scaled_dot_product_attention") else LoRAAttnProcessor LoRAAttnProcessor2_0 if hasattr(F, "scaled_dot_product_attention") else LoRAAttnProcessor
) )
module = lora_attn_processor_class(hidden_size=hidden_size, cross_attention_dim=cross_attention_dim) module = lora_attn_processor_class(hidden_size=hidden_size, cross_attention_dim=cross_attention_dim)
unet_lora_attn_procs[name] = module unet_lora_attn_procs[name] = module
unet_lora_parameters.extend(module.parameters()) unet_lora_parameters.extend(module.parameters())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment