From ee9e209726244df1140e0e616a3d9e23ddf8b13e Mon Sep 17 00:00:00 2001 From: Zeng Xian Date: Sun, 8 Oct 2023 19:27:36 +0800 Subject: [PATCH] fix typo in train dreambooth lora description --- examples/dreambooth/train_dreambooth_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py index 5bb6c78b7b74..47de88f338d1 100644 --- a/examples/dreambooth/train_dreambooth_lora.py +++ b/examples/dreambooth/train_dreambooth_lora.py @@ -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