Create README.md

README.md

@@ -0,0 +1,22 @@
+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
+)