acmyu/KeyframesAI

1accelerate launch --gpu_ids 0,1,2,3,4,5,6,7 --num_processes 8 --use_deepspeed --mixed_precision="fp16"  stage3_train_refined_model.py \2 --pretrained_model_name_or_path="stabilityai/stable-diffusion-2-1-base" \3  --image_encoder_path='facebook/dinov2-giant' \4  --img_path='{image_path}' \5  --json_path='{data.json}' \6  --gen_t_img_path='{stage2_generate}' \7  --output_dir="{output_dir}" \8 --learning_rate=1e-5 \9 --train_batch_size=16 \10 --max_train_steps=1000000 \11 --mixed_precision="fp16" \12 --checkpointing_steps=5000  \13 --noise_offset=0.1 \14 --report_to=tensorboard \15 --lr_warmup_steps 5000  \16