dev
19be1cc9ef
CodeFormer был написан по API сторонней обёртки с PyPI (rohitkhatri): CodeFormer(..., device=..., fp16=False) и net.device официальный класс не принимает — вызов падал бы с TypeError. В photo-ai/vendor/codeformer вендорен официальный sczhou/CodeFormer (codeformer_arch.py + vqgan_arch.py, b33cc7d, лицензия S-Lab 1.0), устройство передаётся через net.to(device), чекпоинт читается из params_ema. Веса больше не качаются лениво при первом запросе: fetch-weights.py на сборке образа кладёт их в /opt/photo-ai-seed (build-arg PHOTO_AI_PREFETCH, дефолт codeformer), при старте seed_weights() переносит их в том photo-ai-models:/models/weights. Том переживает пересборку образа, BuildKit-кэш не даёт качать повторно, недоступная сеть на сборке не роняет образ.
49 lines
2.1 KiB
Docker
49 lines
2.1 KiB
Docker
# syntax=docker/dockerfile:1
|
|
FROM python:3.10-slim
|
|
|
|
ARG TORCH_VARIANT=cpu
|
|
ARG TORCH_INDEX=https://download.pytorch.org/whl/${TORCH_VARIANT}
|
|
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends libgl1 libglib2.0-0 && rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN pip install --no-cache-dir "typing-extensions==4.12.2" "numpy==2.2.6"
|
|
|
|
RUN pip install --no-cache-dir torch torchvision --index-url ${TORCH_INDEX}
|
|
|
|
RUN pip install --no-cache-dir --no-deps basicsr==1.4.2 realesrgan==0.3.0 gfpgan==1.3.8 facexlib==0.3.0 && \
|
|
pip install --no-cache-dir "numpy==2.2.6" "opencv-python-headless==5.0.0.93" addict future lmdb Pillow pyyaml \
|
|
requests scikit-image scipy tqdm filterpy numba fastapi "uvicorn[standard]" python-multipart
|
|
|
|
RUN BASICSR_DEG=$(python -c "import basicsr; import os; print(os.path.join(os.path.dirname(basicsr.__file__), 'data', 'degradations.py'))" 2>/dev/null) || \
|
|
BASICSR_DEG=$(find /usr/local/lib/python3.10 -path "*/basicsr/data/degradations.py" 2>/dev/null | head -1) && \
|
|
if [ -n "$BASICSR_DEG" ]; then \
|
|
sed -i 's/from torchvision.transforms.functional_tensor/from torchvision.transforms.functional/g' "$BASICSR_DEG" && \
|
|
echo "basicsr patch applied to $BASICSR_DEG"; \
|
|
else \
|
|
echo "basicsr degradations.py not found, skipping patch"; \
|
|
fi
|
|
|
|
COPY app.py ./
|
|
COPY fetch-weights.py ./
|
|
COPY vendor/ ./vendor/
|
|
|
|
ENV PHOTO_AI_MODELS_DIR=/models
|
|
ENV MODEL_PATH=/models/RealESRGAN_x2plus.pth
|
|
ENV PHOTO_AI_SEED_DIR=/opt/photo-ai-seed
|
|
|
|
ARG PHOTO_AI_PREFETCH=codeformer
|
|
|
|
RUN --mount=type=cache,target=/var/cache/photo-ai-weights,sharing=locked \
|
|
mkdir -p "$PHOTO_AI_SEED_DIR" && \
|
|
if [ -n "$PHOTO_AI_PREFETCH" ] && [ "$PHOTO_AI_PREFETCH" != "none" ]; then \
|
|
python fetch-weights.py "$PHOTO_AI_SEED_DIR" /var/cache/photo-ai-weights || echo "предзагрузка весов не удалась, сервис скачает их при первом запросе"; \
|
|
else \
|
|
echo "предзагрузка весов отключена (PHOTO_AI_PREFETCH=$PHOTO_AI_PREFETCH)"; \
|
|
fi
|
|
|
|
VOLUME /models
|
|
|
|
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8080"]
|