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
