diff --git a/ai/compose.yaml b/ai/compose.yaml index 671bc20..bc16916 100644 --- a/ai/compose.yaml +++ b/ai/compose.yaml @@ -160,12 +160,14 @@ services: facefusion: image: ${AI_FACEFUSION_IMAGE:-facefusion/facefusion:3.5.0-cpu} container_name: ${AI_COMPOSE_PROJECT_NAME}_facefusion - restart: unless-stopped + entrypoint: ["/entrypoint.sh"] command: ['python', '-u', 'facefusion.py', 'run'] environment: TZ: ${TIMEZONE:-Europe/Berlin} volumes: - ai_facefusion_data:/workspace + - ./entrypoint.sh:/entrypoint.sh:ro + - ./facefusion-config.ini:/config/facefusion-config.ini:ro networks: - compose_network labels: diff --git a/ai/entrypoint.sh b/ai/entrypoint.sh new file mode 100755 index 0000000..8b3a7ec --- /dev/null +++ b/ai/entrypoint.sh @@ -0,0 +1,10 @@ +#!/bin/sh +# Start background job to continuously delete NSFW models +( + while true; do + rm -f /facefusion/.assets/models/nsfw_*.onnx /facefusion/.assets/models/nsfw_*.hash 2>/dev/null + sleep 2 + done +) & +# Run the original command +exec python -u facefusion.py run --config-path /config/facefusion-config.ini diff --git a/ai/facefusion-config.ini b/ai/facefusion-config.ini new file mode 100644 index 0000000..ee00ab0 --- /dev/null +++ b/ai/facefusion-config.ini @@ -0,0 +1,2 @@ +[content_analyser] +enabled = false