Merge pull request #30 from jkremser/use_safety_checker

stable-diff: Don't allow to create nsfw content
kedify · Jun 19, 2024 · c91b721 · c91b721
2 parents f4f6fb4 + 22b4452
commit c91b721
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/samples/stable-diffusion/tweaks/consumer.sh b/samples/stable-diffusion/tweaks/consumer.sh
@@ -25,7 +25,7 @@ generate() {
   _prompt=$(echo ${_imageRequest} | jq '.prompt')
   _count=$(echo ${_imageRequest} | jq 2> /dev/null '.count // 1')
   touch ${FLAG_FILE}
-  python /app/src/app.py --number_of_images "${_count}" --prompt "${_prompt}"
+  python /app/src/app.py --use_safety_checker --number_of_images "${_count}" --prompt "${_prompt}"
   rm -rf ${FLAG_FILE}
   echo "Done. Image for ${_imageRequest} has been stored in /app/results."
   sleep 1