|
export MODEL_NAME="runwayml/stable-diffusion-v1-5" |
|
export OUTPUT_DIR="/home/long.qul/tritontest" |
|
export DATASET_NAME="lambdalabs/pokemon-blip-captions" |
|
|
|
accelerate launch --mixed_precision="fp16" lora_test_1.py \ |
|
--pretrained_model_name_or_path=$MODEL_NAME \ |
|
--dataset_name=$DATASET_NAME \ |
|
--dataloader_num_workers=8 \ |
|
--resolution=512 --center_crop --random_flip \ |
|
--train_batch_size=1 \ |
|
--gradient_accumulation_steps=4 \ |
|
--max_train_steps=15000 \ |
|
--learning_rate=1e-04 \ |
|
--max_grad_norm=1 \ |
|
--lr_scheduler="cosine" --lr_warmup_steps=0 \ |
|
--output_dir=${OUTPUT_DIR} \ |
|
--push_to_hub \ |
|
--hub_model_id=${HUB_MODEL_ID} \ |
|
--checkpointing_steps=500 \ |
|
--validation_prompt="A pokemon with blue eyes." \ |
|
--seed=1337 |