diff --git a/Dockerfile b/Dockerfile index 9e34c2a..d062bff 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ ARG VARIANT=cpu # ---- CPU base ---- -FROM python:3.12-slim AS base-cpu +FROM python:3.11-slim AS base-cpu RUN apt-get update && apt-get install -y --no-install-recommends \ curl libgl1 libglib2.0-0 \ @@ -22,9 +22,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ && add-apt-repository ppa:deadsnakes/ppa \ && apt-get update \ && apt-get install -y --no-install-recommends \ - python3.12 python3.12-venv python3.12-dev \ + python3.11 python3.11-venv python3.11-dev \ curl libgl1 libglib2.0-0 \ - && ln -sf /usr/bin/python3.12 /usr/bin/python3 \ + && ln -sf /usr/bin/python3.11 /usr/bin/python3 \ && ln -sf /usr/bin/python3 /usr/bin/python \ && python3 -m ensurepip --upgrade \ && python3 -m pip install --no-cache-dir --upgrade pip \ diff --git a/requirements-cpu.txt b/requirements-cpu.txt index 9c9a466..fb0d321 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -1,4 +1,4 @@ -torch==2.10.0 -torchvision==0.15.0 -Real-ESRGAN==0.3.0 -basicsr==1.4.2 +torch==2.0.1 +torchvision==0.15.2 +realesrgan>=0.2.5 +basicsr>=1.4.1 diff --git a/requirements-gpu.txt b/requirements-gpu.txt index bd9b8a2..fbf8356 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -1,5 +1,5 @@ ---extra-index-url https://download.pytorch.org/whl/cu124 -torch==2.5.1 -torchvision==0.20.1 -Real-ESRGAN==0.3.0 -basicsr==1.4.2 +--extra-index-url https://download.pytorch.org/whl/cu118 +torch==2.0.1 +torchvision==0.15.2 +realesrgan>=0.2.5 +basicsr>=1.4.1 diff --git a/requirements.txt b/requirements.txt index 769056d..d5efc1f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,12 @@ -fastapi==0.115.6 -uvicorn[standard]==0.34.0 -python-multipart==0.0.18 -pydantic-settings==2.7.1 -psutil==6.1.1 -httpx==0.27.0 +fastapi==0.104.1 +uvicorn[standard]==0.24.0 +python-multipart==0.0.6 +pydantic-settings==2.1.0 +psutil==5.9.6 +httpx==0.25.2 # Real-ESRGAN and dependencies -opencv-python==4.10.0.84 -numpy==1.26.4 -scipy==1.14.1 -tqdm==4.67.1 +opencv-python==4.8.1.78 +numpy==1.24.3 +scipy==1.11.4 +tqdm==4.66.1