diff --git a/docs/source/en/training/text2image.md b/docs/source/en/training/text2image.md index f69e9a710e8f..c8abb189a91f 100644 --- a/docs/source/en/training/text2image.md +++ b/docs/source/en/training/text2image.md @@ -181,7 +181,7 @@ accelerate launch --mixed_precision="fp16" train_text_to_image.py \ --max_train_steps=15000 \ --learning_rate=1e-05 \ --max_grad_norm=1 \ - --enable_xformers_memory_efficient_attention + --enable_xformers_memory_efficient_attention \ --lr_scheduler="constant" --lr_warmup_steps=0 \ --output_dir="sd-naruto-model" \ --push_to_hub