stable-diff: Don't allow to create nsfw content

Signed-off-by: Jirka Kremser <[email protected]>
kedify · Jun 19, 2024 · 22b4452 · 22b4452
1 parent f4f6fb4
commit 22b4452
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