Siddharth Kumar Sah 29a382e9e0 feat: add GPU/CUDA acceleration support (:cuda Docker tag) ...

Add a :cuda Docker image tag that auto-detects NVIDIA GPU at runtime
and falls back gracefully to CPU. Same pattern as Immich.

- New gpu.py shared utility for cached CUDA detection
- Background removal (rembg): pass CUDAExecutionProvider to ONNX Runtime
- Upscaling (Real-ESRGAN): use CUDA device + FP16 when GPU available
- OCR (PaddleOCR): enable use_gpu when CUDA detected
- Dispatcher reports GPU status at startup via readiness signal
- Admin health endpoint exposes GPU availability
- Dockerfile uses ARG GPU=false with conditional NVIDIA CUDA base image
- docker-compose.gpu.yml override for GPU users
- CI/CD workflows build and publish :cuda tag (amd64 only)

Three tags: :latest (CPU), :lite (no AI), :cuda (GPU with CPU fallback)

2026-04-05 19:12:45 +08:00

..

detect_faces.py

fix(blur-faces): switch from MediaPipe to OpenCV and auto-orient images

2026-03-26 16:01:56 +08:00

dispatcher.py

feat: add GPU/CUDA acceleration support (:cuda Docker tag)

2026-04-05 19:12:45 +08:00

gpu.py

feat: add GPU/CUDA acceleration support (:cuda Docker tag)

2026-04-05 19:12:45 +08:00

inpaint.py

feat(erase-object): replace mask upload with in-browser brush painting

2026-03-26 17:33:17 +08:00

ocr.py

feat: add GPU/CUDA acceleration support (:cuda Docker tag)

2026-04-05 19:12:45 +08:00

remove_bg.py

feat: add GPU/CUDA acceleration support (:cuda Docker tag)

2026-04-05 19:12:45 +08:00

requirements-gpu.txt

feat: add GPU/CUDA acceleration support (:cuda Docker tag)

2026-04-05 19:12:45 +08:00

requirements.txt

feat: add image-engine and ai stub packages

2026-03-22 02:44:46 +08:00

upscale.py

feat: add GPU/CUDA acceleration support (:cuda Docker tag)

2026-04-05 19:12:45 +08:00