[#15] Use requirements_minimal.txt as dependencies

.github/workflows/build-image.yaml

@@ -7,6 +7,9 @@ on:
 
 jobs:
   Build-Docker-Image:
+    strategy:
+      matrix:
+        demucs_version: [ minimal ]
     runs-on: ubuntu-latest
     steps:
       -
@@ -39,6 +42,7 @@ jobs:
         uses: docker/build-push-action@v3
         with:
           context: .
+          build-args: DEMUCS_VERSION=${{ matrix.demucs_version }}
           push: true
           platforms: linux/amd64,linux/arm64
           tags: xserrat/facebook-demucs:${{ steps.github_ref_name_replacement_step.outputs.replaced }}

Dockerfile

@@ -1,19 +1,8 @@
-#FROM python:3.8-alpine3.16 as build-lameenc
-#
-#RUN python3 -m venv /opt/venv  \
-#    && python3 -m pip install --upgrade pip
-#ENV PATH="/opt/venv/bin:$PATH"
-#
-#RUN git clone -b v1.3.1 --single-branch https://github.com/chrisstaite/lameenc /lib/lameenc
-#
-#RUN mkdir -p /lib/lameenc/build
-#WORKDIR /lib/lameenc/build
-#RUN cmake .. && make && pip install "lameenc-1.3.1-cp38-cp38-linux_aarch64.whl"
-
 FROM python:3.8-alpine3.16
 
 USER root
 ENV TORCH_HOME=/data/models
+ENV REQUIREMENTS_FILE=requirements_minimal.txt
 
 # Install needed packages
 RUN apk update && apk add \
@@ -27,12 +16,11 @@ WORKDIR /lib/demucs
 
 RUN git clone --depth 1 --branch main https://github.com/facebookresearch/demucs .
 
-#COPY --from=build-lameenc /opt/venv /opt/venv
+#RUN REQUIREMENTS_FILE=$([[ "${DEMUCS_VERSION}" == "minimal" ]] && echo "requirements_minimal.txt" || echo "requirements.txt")
 
-RUN sed -i 's/lameenc>=1.2/lameenc>=1.4.1/g' requirements.txt requirements_minimal.txt  \
-    && cat requirements.txt \
-    && cat requirements_minimal.txt
-RUN python3 -m pip install --platform=arm64 --no-deps -e .
+RUN echo "Requirements file: ${REQUIREMENTS_FILE}"
+RUN sed -i 's/lameenc>=1.2/lameenc>=1.4.1/g' ${REQUIREMENTS_FILE} # Upgrade lameenc to 1.4.1 to support arm64
+RUN python3 -m pip install -r ${REQUIREMENTS_FILE}
 RUN python3 -m demucs.separate -d cpu --mp3 test.mp3 # Trigger model download \
     && rm -r separated  # cleanup