summaryrefslogtreecommitdiffstats
path: root/train_dreambooth.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r--train_dreambooth.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py
index 7745d27..d284346 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -262,7 +262,7 @@ def parse_args():
262 ) 262 )
263 parser.add_argument( 263 parser.add_argument(
264 "--text_encoder_unfreeze_last_n_layers", 264 "--text_encoder_unfreeze_last_n_layers",
265 default=2, 265 default=-1,
266 help="Number of text encoder layers to train.", 266 help="Number of text encoder layers to train.",
267 ) 267 )
268 parser.add_argument( 268 parser.add_argument(