add codes

README.md

@@ -21,7 +21,7 @@ accelerate launch \
     --instance_data_dir="./nyc_ads_dataset" \
     --instance_prompt="a photo of an urbanad nyc" \
     --output_dir="./nyc-ad-model" \
-    --resolution=512 \
+    --resolution=100 \
     --train_batch_size=1 \
     --gradient_accumulation_steps=1 \
     --gradient_checkpointing \
@@ -35,4 +35,10 @@ accelerate launch \
     --report_to="tensorboard" \
     --logging_dir="./nyc-ad-model/logs" 
 
-git clone https://github.com/huggingface/diffusers.git
+git clone https://github.com/huggingface/diffusers.git
+
+export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
+
+import torch
+torch.cuda.empty_cache()
+torch.cuda.reset_peak_memory_stats()

xformers

@@ -0,0 +1 @@
+Subproject commit 8fc8ec5a4d6498ff81c0c418b89bbaf133ae3a44