diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 2 |
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( |