mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
c5376c5695
* add the controlnet pipeline for pixart alpha --------- Co-authored-by: YiYi Xu <yixu310@gmail.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: junsongc <cjs1020440147@icloud.com>
24 lines
587 B
Bash
Executable File
24 lines
587 B
Bash
Executable File
#!/bin/bash
|
|
|
|
# run
|
|
# accelerate config
|
|
|
|
# check with
|
|
# accelerate env
|
|
|
|
export MODEL_DIR="PixArt-alpha/PixArt-XL-2-512x512"
|
|
export OUTPUT_DIR="output/pixart-controlnet-hf-diffusers-test"
|
|
|
|
accelerate launch ./train_pixart_controlnet_hf.py --mixed_precision="fp16" \
|
|
--pretrained_model_name_or_path=$MODEL_DIR \
|
|
--output_dir=$OUTPUT_DIR \
|
|
--dataset_name=fusing/fill50k \
|
|
--resolution=512 \
|
|
--learning_rate=1e-5 \
|
|
--train_batch_size=1 \
|
|
--gradient_accumulation_steps=4 \
|
|
--report_to="wandb" \
|
|
--seed=42 \
|
|
--dataloader_num_workers=8
|
|
# --lr_scheduler="cosine" --lr_warmup_steps=0 \
|