From 937bac68fb5d45fbba8ea1d9fd1403b1c0b18041 Mon Sep 17 00:00:00 2001
From: Wang Qiang <37444407+wangqiang9@users.noreply.github.com>
Date: Wed, 18 Sep 2024 23:14:14 +0800
Subject: [PATCH] Fix train.py: error: unrecognized arguments:
 --enable_xformers_memory_efficient_attention (#6)

Fix train.py: error: unrecognized arguments: --enable_xformers_memory_efficient_attention
---
 scripts/README_TRAIN.md | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/scripts/README_TRAIN.md b/scripts/README_TRAIN.md
index 3a16ef2..93b6fbc 100644
--- a/scripts/README_TRAIN.md
+++ b/scripts/README_TRAIN.md
@@ -34,7 +34,6 @@ accelerate launch --mixed_precision="bf16" scripts/train.py \
   --lr_warmup_steps=100 \
   --seed=42 \
   --output_dir="output_dir" \
-  --enable_xformers_memory_efficient_attention \
   --gradient_checkpointing \
   --mixed_precision="bf16" \
   --adam_weight_decay=3e-2 \
@@ -81,7 +80,6 @@ accelerate launch --use_deepspeed --deepspeed_config_file config/zero_stage2_con
   --lr_warmup_steps=100 \
   --seed=42 \
   --output_dir="output_dir" \
-  --enable_xformers_memory_efficient_attention \
   --gradient_checkpointing \
   --mixed_precision="bf16" \
   --adam_weight_decay=3e-2 \
@@ -98,4 +96,4 @@ accelerate launch --use_deepspeed --deepspeed_config_file config/zero_stage2_con
   --train_mode="inpaint" \
   --resume_from_checkpoint="latest" \
   --trainable_modules "."
-```
\ No newline at end of file
+```