update requirement basicsr

Dockerfile

@@ -29,54 +29,23 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-# RUN useradd -m -u 1000 user
-# USER user
-# ENV HOME=/home/user \
-#     PATH=/home/user/.local/bin:${PATH}
-# WORKDIR ${HOME}/app
-
-# RUN curl https://pyenv.run | bash
-# ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
-# ENV PYTHON_VERSION=3.10.18
-# RUN pyenv install ${PYTHON_VERSION} && \
-#     pyenv global ${PYTHON_VERSION} && \
-#     pyenv rehash && \
-#     pip install --no-cache-dir -U pip setuptools wheel
-
-# RUN pip install --user --no-cache-dir -U torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124
-# RUN pip install --user --no-cache-dir -U f5_tts==1.1.9
-# COPY --chown=1000 requirements_f5tts.txt /tmp/requirements_f5tts.txt
-# RUN pip install --user --no-cache-dir -U -r /tmp/requirements_f5tts.txt
-# COPY --chown=1000 requirements.txt /tmp/requirements.txt
-# RUN pip install --user --no-cache-dir -U -r /tmp/requirements.txt
-
-# COPY --chown=1000 . ${HOME}/app
-# RUN ls -a
-# ENV PYTHONPATH=${HOME}/app \
-#     PYTHONUNBUFFERED=1 \
-#     GRADIO_ALLOW_FLAGGING=never \
-#     GRADIO_NUM_PORTS=1 \
-#     GRADIO_SERVER_NAME=0.0.0.0 \
-#     GRADIO_THEME=huggingface \
-#     SYSTEM=spaces
-# CMD ["python", "app.py"]
-
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
+    PATH=/home/user/.local/bin:${PATH}
 WORKDIR ${HOME}/app
 
 RUN curl https://pyenv.run | bash
-ENV PYTHON_VERSION=3.10.18
+ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
+ENV PYTHON_VERSION=3.10.9
 RUN pyenv install ${PYTHON_VERSION} && \
     pyenv global ${PYTHON_VERSION} && \
     pyenv rehash && \
-    pyenv exec pip install --no-cache-dir -U pip setuptools wheel && \
-    pyenv exec pip install --no-cache-dir -U torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124 && \
-    pyenv exec pip install --no-cache-dir -U f5_tts==1.1.9 && \
-    # pyenv exec pip install --no-cache-dir -U -r /tmp/requirements_f5tts.txt && \
-    pyenv exec pip install --no-cache-dir -U -r /tmp/requirements.txt
+    pip install --no-cache-dir -U pip setuptools wheel
+
+RUN pip install --no-cache-dir -U torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0
+COPY --chown=1000 requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
 
 COPY --chown=1000 . ${HOME}/app
 RUN ls -a

requirements.txt

@@ -1,6 +1,6 @@
-# torch==2.4.0
-# torchaudio==2.4.0
-# torchvision==0.19.0
+torch==2.4.0
+torchaudio==2.4.0
+torchvision==0.19.0
 
 face_alignment==1.3.0
 imageio==2.19.3
@@ -16,7 +16,7 @@ yacs==0.1.8
 pyyaml
 joblib==1.1.0
 scikit-image==0.19.3
-basicsr==1.4.2
+basicsr @ git+https://github.com/XPixelGroup/BasicSR.git@8d56e3a045f9fb3e1d8872f92ee4a4f07f886b0a
 facexlib
 dlib-bin
 gfpgan
@@ -44,5 +44,8 @@ transformers_stream_generator
 vocos
 wandb
 x_transformers>=1.31.14
+f5-tts
 
 spaces
+soxr
+ffmpeg