Update README.md

2026-07-26 22:30:28 +00:00 · 2024-07-19 03:32:01 -05:00
parent 68a54f4ed8
commit 25d30be5e1
1 changed files with 2 additions and 1 deletions
@@ -63,7 +63,8 @@ python -m torch.distributed.launch --nnodes=1 --nproc_per_node=8 train_deepspeed
 --model DiT-XL/2 \
 --num_experts 8 \
 --num_experts_per_tok 2 \
--data-path /maindata/data/shared/multimodal/public/dataset_img_only/imagenet/data/train \
+--data-path /path/to/imagenet/train \
+--vae-path /path/to/vae \
 --train_batch_size 32
 ```