From c29c837728b4e0d744fbeec11de6901b65862ffd Mon Sep 17 00:00:00 2001
From: Xavier Serrat Bordas <xavier.serrat22@gmail.com>
Date: Thu, 17 Nov 2022 00:30:00 +0100
Subject: [PATCH] test arm

---
 Dockerfile | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index a3e23c3..709bce6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,3 +1,15 @@
+#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
@@ -6,8 +18,7 @@ ENV TORCH_HOME=/data/models
 # Install needed packages
 RUN apk update && apk add \
     git \
-    ffmpeg \
-    && python3 -m pip install --upgrade pip
+    ffmpeg
 
 # Install Facebook Demucs
 RUN mkdir -p /lib/demucs
@@ -16,10 +27,12 @@ WORKDIR /lib/demucs
 
 RUN git clone --depth 1 --branch main https://github.com/facebookresearch/demucs .
 
+#COPY --from=build-lameenc /opt/venv /opt/venv
+
 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 -e .
+RUN python3 -m pip install --platform=arm64 --no-deps -e .
 RUN python3 -m demucs.separate -d cpu --mp3 test.mp3 # Trigger model download \
     && rm -r separated  # cleanup