Skip to content

Commit 4a3e574

Browse files
make style
1 parent c2a28c3 commit 4a3e574

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/dreambooth/train_dreambooth_lora.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -871,7 +871,7 @@ def main(args):
871871
lora_attn_processor_class = (
872872
LoRAAttnProcessor2_0 if hasattr(F, "scaled_dot_product_attention") else LoRAAttnProcessor
873873
)
874-
874+
875875
module = lora_attn_processor_class(hidden_size=hidden_size, cross_attention_dim=cross_attention_dim)
876876
unet_lora_attn_procs[name] = module
877877
unet_lora_parameters.extend(module.parameters())

0 commit comments

Comments
 (0)