| args = Namespace( | |
| image_path="/content/images/Training Images", | |
| prompt_path="/content/data.csv", | |
| output_path="/content/output", | |
| model="CompVis/stable-diffusion-v1-4", | |
| seed=1703, | |
| mixed_precision="bf16", | |
| rank=4, | |
| use_rslora=True, | |
| init_lora_weights="gaussian", | |
| lr=1e-5, | |
| gradient_accumulation_steps=1, | |
| train_batch_size=64, | |
| num_train_epochs=20, | |
| lr_scheduler="constant", | |
| lr_warmup_steps=1, | |
| snr_gamma=5, | |
| size=268, | |
| max_grad_norm=1, | |
| use_8bit_adam=True, | |
| checkpointing_steps=1100, | |
| resume_from_checkpoint=None | |
| ) |