| args = Namespace( | |
| model = "kandinsky-community/kandinsky-2-2-prior", | |
| image_path = "/content/images/CFP", | |
| prompt_path = "/content/data.csv", | |
| output_path = "/content/output", | |
| mixed_precision='bf16', | |
| gradient_accumulation_steps=1, | |
| seed = 1703, | |
| rank=32, | |
| use_rslora=True, | |
| init_lora_weights='gaussian', | |
| max_grad_norm=1, | |
| lr=1e-4, | |
| lr_scheduler='constant', | |
| lr_warmup_steps=0, | |
| train_batch_size=32, | |
| num_train_epochs=10, | |
| snr_gamma=5, | |
| resume_from_checkpoint=None, | |
| checkpointing_steps=680 | |
| ) |