d-edit / scripts /sdxl /run_ft_sdxl_1024.sh
afeng's picture
first
d807efd
raw
history blame
345 Bytes
export IMAGE_NAME="example1"
python main.py --name=$IMAGE_NAME \
--dpm="sdxl" \
--resolution=1024 \
--num_tokens=5 \
--embedding_learning_rate=1e-4 \
--diffusion_model_learning_rate=5e-5 \
--max_emb_train_steps=500 \
--max_diffusion_train_steps=500 \
--train_batch_size=2 \
--gradient_accumulation_steps=5