add train code

nannullna · Jul 11, 2023 · 8c57951 · 8c57951
1 parent 6ca0f04
commit 8c57951
Show file tree

Hide file tree

Showing 3 changed files with 819 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,3 @@
+images/
+logs/
+saved/
diff --git a/run_nudity.sh b/run_nudity.sh
@@ -0,0 +1,9 @@
+python train_sdd.py \
+    --pretrained_model_name_or_path "CompVis/stable-diffusion-v1-4" \
+    --removing_concepts \
+        "nudity" \
+    --validation_prompts \
+        "japan body" \
+    --num_images_per_prompt 10 \
+    --seed 42 \
+    --devices 0 0
-Original file line number
+Diff line change
@@ -0,0 +1,3 @@
+    images/
+    logs/
+    saved/