| args = Namespace( |
| image_path="/content/images/CFP", |
| output_path="/content/output", |
| prior_model="kandinsky-community/kandinsky-2-2-prior", |
| decoder_model="kandinsky-community/kandinsky-2-2-decoder", |
| gradient_accumulation_steps=1, |
| mixed_precision='bf16', |
| seed=1703, |
| size=256, |
| 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 |
| ) |
