local dockerfile working

argilla-io · May 14, 2024 · 4721b7a · 4721b7a
1 parent 00a2e43
commit 4721b7a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/docker/local.Dockerfile b/docker/local.Dockerfile
@@ -1,8 +1,18 @@
+FROM python:3.11-slim AS base
+
+RUN apt-get update && \
+    apt-get upgrade -y && \
+    apt-get autoremove -y && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+WORKDIR /
+
 FROM python:3.11-slim 
 
-WORKDIR /app
+COPY . .
 
-RUN pip install "distilabel[argilla]"
+RUN pip install -e ".[argilla,hf-transformers,hf-inference-endpoints]"
 
 EXPOSE 80