add hakubooru dataset utils

caradryanl · May 15, 2024 · d76cbd0 · d76cbd0
1 parent a029083
commit d76cbd0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -34,4 +34,5 @@ models/diffusers/*
 diffusers/outputs/*
 !diffusers/experiments/secmi_random/*
 !diffusers/experiments/pia/*
-diffusers/datasets/*
+diffusers/datasets/*
+!diffusers/datasets/placeholder.txt
diff --git a/requirements.txt b/requirements.txt
@@ -27,3 +27,10 @@ scikit-learn
 # by drc:
 opencv-python
 scikit-image
+
+# by hakubooru
+peewee
+webdataset
+python-dateutil
+pytorch_lightning
+hakubooru