| 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 | |
| ) |