Add explicit HYPIR restoration profile

compose.yaml

@@ -571,6 +571,7 @@ services:
       CONTROLLER_TOKEN: "${CONTROLLER_TOKEN:?CONTROLLER_TOKEN is required}"
       ALLOWED_PROFILES: fast,medium,beta1,large,ultra,uncensored
       IMAGE_WORKER: image
+      RESTORE_WORKER: restore
       TTS_WORKER: qwen3
     networks: [control]
     security_opt: ["no-new-privileges:true"]
@@ -617,6 +618,11 @@ 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"
@@ -693,6 +699,41 @@ 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
+      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