Skip to content

GitLab

Unverified Commit 0513a8cf authored by Zeng Xian's avatar Zeng Xian Committed by GitHub
Browse files

fix typo in train dreambooth lora description (#5332)

parent 306dc6e7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/dreambooth/train_dreambooth_lora.py
View file @ 0513a8cf
......@@ -854,7 +854,7 @@ def main(args):
# For Stable Diffusion, it should be equal to:
# - down blocks (2x attention layers) * (2x transformer layers) * (3x down blocks) = 12
# - mid blocks (2x attention layers) * (1x transformer layers) * (1x mid blocks) = 2
# - up blocks (2x attention layers) * (3x transformer layers) * (3x down blocks) = 18
# - up blocks (2x attention layers) * (3x transformer layers) * (3x up blocks) = 18
# => 32 layers
# Set correct lora layers
......
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