Update train.sh

examples/train/train.sh

-
+  python -m src.train.distill \
+  --pretrained_model_name_or_path="stabilityai/stable-diffusion-2-1" \
+  --mixed_precision="bf16" \
+  --resolution=768 \
+  --train_batch_size=6 \
+  --gradient_accumulation_steps=2 \
+  --dataloader_num_workers=6 \
+  --learning_rate=1e-04 \
+  --weight_decay=0. \
+  --output_dir="ckpt/linfusion_sd2p1" \
+  --save_steps=10000