nodes/aigc-apps-VideoX-Fun
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix train.py: error: unrecognized arguments: --enable_xformers_memory_efficient_attention (#6)

Browse Source 
Fix train.py: error: unrecognized arguments: --enable_xformers_memory_efficient_attention
This commit is contained in:
Wang Qiang
2024-09-18 23:14:14 +08:00
committed by GitHub
parent 0323907db5
commit 937bac68fb
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/README_TRAIN.md
+1 -3
View File
@@ -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 "."
```
```