chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
accelerate launch train_textdiffuser2_t2i_lora.py \
|
||||
--pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5" \
|
||||
--train_batch_size=18 \
|
||||
--gradient_accumulation_steps=4 \
|
||||
--gradient_checkpointing \
|
||||
--mixed_precision="fp16" \
|
||||
--num_train_epochs=6 \
|
||||
--learning_rate=1e-4 \
|
||||
--text_encoder_learning_rate=1e-5 \
|
||||
--lr_scheduler="constant" \
|
||||
--output_dir="diffusion_experiment_result" \
|
||||
--enable_xformers_memory_efficient_attention \
|
||||
--dataloader_num_workers=8 \
|
||||
--index_file_path='/path/to/train_dataset_index.txt' \
|
||||
--dataset_path='/path/to/laion-ocr-select/' \
|
||||
--granularity=128 \
|
||||
--coord_mode="ltrb" \
|
||||
--max_length=77 \
|
||||
--resume_from_checkpoint="latest"
|
||||
Reference in New Issue
Block a user