Create README.md

README.md

@@ -0,0 +1,23 @@
+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=40,
+    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
+)