From 95b9407a3e1f13f8b3f71dfb2e9efa236d7a1567 Mon Sep 17 00:00:00 2001
From: Maw-Fox <huskyz0rz@gmail.com>
Date: Sat, 12 Nov 2022 18:48:16 -0700
Subject: [PATCH] Add+config .gitignore (bring back git stage) and fix up
 documentation.

---
 .gitignore                   | 10 ++++++++++
 trainer/diffusers_trainer.py |  4 ++--
 2 files changed, 12 insertions(+), 2 deletions(-)
 create mode 100644 .gitignore

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..a454386
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,10 @@
+aesthetic/**
+*_cropped/
+*_cropped/**
+**/nsfw-ids.txt
+**/*.image
+**/*.caption
+**/dataset*.tar
+*/**/*.json
+**/*.png
+**/*.jpg
diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py
index 33c522c..2c6072c 100644
--- a/trainer/diffusers_trainer.py
+++ b/trainer/diffusers_trainer.py
@@ -2,8 +2,8 @@
 # `nvcc --version` to get CUDA version.
 # `pip install -i https://test.pypi.org/simple/ bitsandbytes-cudaXXX` to install for current CUDA.
 # Example Usage:
-# Single GPU: torchrun --nproc_per_node=1 trainer_dist.py --model="CompVis/stable-diffusion-v1-4" --run_name="liminal" --dataset="liminal-dataset" --hf_token="hf_blablabla" --bucket_side_min=64 --use_8bit_adam=True --gradient_checkpointing=True --batch_size=10 --fp16=True --image_log_steps=250 --epochs=20 --resolution=768 --use_ema=True
-# Multiple GPUs: torchrun --nproc_per_node=N trainer_dist.py --model="CompVis/stable-diffusion-v1-4" --run_name="liminal" --dataset="liminal-dataset" --hf_token="hf_blablabla" --bucket_side_min=64 --use_8bit_adam=True --gradient_checkpointing=True --batch_size=10 --fp16=True --image_log_steps=250 --epochs=20 --resolution=768 --use_ema=True
+# Single GPU: torchrun --nproc_per_node=1 trainer/diffusers_trainer.py --model="CompVis/stable-diffusion-v1-4" --run_name="liminal" --dataset="liminal-dataset" --hf_token="hf_blablabla" --bucket_side_min=64 --use_8bit_adam --gradient_checkpointing --batch_size=1 --fp16 --image_log_steps=250 --epochs=20 --resolution=768 --use_ema
+# Multiple GPUs: torchrun --nproc_per_node=N trainer/diffusers_trainer.py --model="CompVis/stable-diffusion-v1-4" --run_name="liminal" --dataset="liminal-dataset" --hf_token="hf_blablabla" --bucket_side_min=64 --use_8bit_adam --gradient_checkpointing --batch_size=10 --fp16 --image_log_steps=250 --epochs=20 --resolution=768 --use_ema
 
 import argparse
 import socket