Remove ineffective photo restoration pipeline
This commit is contained in:
@@ -618,11 +618,6 @@ services:
|
||||
IMAGE_WORKER_URL: http://image-worker:8086
|
||||
IMAGE_WORKER_TOKEN: "${CONTROLLER_TOKEN:?CONTROLLER_TOKEN is required}"
|
||||
IMAGE_MODEL_NAME: FLUX.2-klein-9B-fp8-beta
|
||||
RESTORATION_WORKER_URL: http://restoration-worker:8087
|
||||
RESTORATION_WORKER_TOKEN: "${CONTROLLER_TOKEN:?CONTROLLER_TOKEN is required}"
|
||||
RESTORATION_MODEL_NAME: HYPIR-SD2
|
||||
RESTORATION_CHAT_MODEL: restauration
|
||||
RESTORATION_CHAT_PROFILE: fast
|
||||
CHAT_IMAGE_ALLOW_REMOTE_URLS: "false"
|
||||
ENABLE_IMAGE_GENERATION: "true"
|
||||
ENABLE_TTS: "true"
|
||||
@@ -699,44 +694,6 @@ services:
|
||||
timeout: 3s
|
||||
retries: 12
|
||||
|
||||
restoration-worker:
|
||||
build:
|
||||
context: platform/docker/restoration-worker
|
||||
args:
|
||||
HYPIR_COMMIT: b61d107c6cef38f01a93c7833558869731cfa8c1
|
||||
image: mike-ai/restoration-worker:hypir-sd2
|
||||
container_name: mike-ai-restoration-worker
|
||||
restart: "no"
|
||||
profiles: [image]
|
||||
labels:
|
||||
com.mike-ai.image-worker: restore
|
||||
gpus: all
|
||||
read_only: true
|
||||
tmpfs: ["/tmp:size=1g,mode=1777"]
|
||||
volumes:
|
||||
- "${HYPIR_MODEL_DIR:-/data/models/HYPIR}:/models/hypir:ro"
|
||||
- "${HYPIR_SD2_DIR:-/data/models/stable-diffusion-2-1-base}:/models/sd2:ro"
|
||||
- router-images:/data/images
|
||||
environment:
|
||||
NVIDIA_VISIBLE_DEVICES: "${RESTORATION_GPU_DEVICE:-1}"
|
||||
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
||||
WORKER_TOKEN: "${CONTROLLER_TOKEN:?CONTROLLER_TOKEN is required}"
|
||||
HYPIR_BASE_MODEL: /models/sd2
|
||||
HYPIR_WEIGHT_FILE: /models/hypir/HYPIR_sd2.pth
|
||||
HYPIR_DEVICE: cuda:0
|
||||
# Empty conditioning avoids prompt-driven repainting in the conservative
|
||||
# restoration profile. Set to passthrough only for deliberate experiments.
|
||||
HYPIR_PROMPT_MODE: "empty"
|
||||
IMAGE_DIR: /data/images
|
||||
networks: [inference]
|
||||
security_opt: ["no-new-privileges:true"]
|
||||
cap_drop: [ALL]
|
||||
healthcheck:
|
||||
test: [CMD, python, -c, "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8087/health', timeout=2)"]
|
||||
interval: 5s
|
||||
timeout: 3s
|
||||
retries: 12
|
||||
|
||||
piper:
|
||||
build:
|
||||
context: platform/docker/piper
|
||||
|
||||
Reference in New Issue
Block a user