add file

README.md

@@ -10,4 +10,30 @@ pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+commands:
+
+accelerate launch \
+  --deepspeed_config_file ds_config.json \
+  diffusers/examples/dreambooth/train_dreambooth.py \
+    --pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5" \
+    --instance_data_dir="./nyc_ads_dataset" \
+    --instance_prompt="a photo of an urbanad nyc" \
+    --output_dir="./nyc-ad-model" \
+    --resolution=512 \
+    --train_batch_size=1 \
+    --gradient_accumulation_steps=1 \
+    --gradient_checkpointing \
+    --learning_rate=5e-6 \
+    --lr_scheduler="constant" \
+    --lr_warmup_steps=0 \
+    --max_train_steps=400 \
+    --mixed_precision="fp16" \
+    --checkpointing_steps=100 \
+    --checkpoints_total_limit=1 \
+    --report_to="tensorboard" \
+    --logging_dir="/content/nyc-ad-model/logs" \
+    --enable_xformers_memory_efficient_attention
+
+git clone https://github.com/huggingface/diffusers.git

diffusers

@@ -0,0 +1 @@
+Subproject commit 0ac1d5b4822e7622da26e7ee11effdb869cd4989

ds_config.json

@@ -0,0 +1,20 @@
+{
+    "zero_optimization": {
+      "stage": 2,
+      "offload_param": {
+        "device": "cpu",
+        "pin_memory": true
+      },
+      "offload_optimizer": {
+        "device": "cpu",
+        "pin_memory": true
+      },
+      "overlap_comm": true,
+      "contiguous_gradients": true
+    },
+    "gradient_accumulation_steps": 1,
+    "train_batch_size": 1,
+    "fp16": {
+      "enabled": true
+    }
+}