add esd training code

nannullna · Jul 11, 2023 · 046a493 · 046a493
1 parent 5e0b378
commit 046a493
Show file tree

Hide file tree

Showing 4 changed files with 829 additions and 10 deletions.
diff --git a/run_esd_nudity.sh b/run_esd_nudity.sh
@@ -0,0 +1,9 @@
+python train_esd.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
diff --git a/run_nudity.sh → run_sdd_nudity.sh b/run_nudity.sh → run_sdd_nudity.sh