Browse Source

run exo in docker

Alex Cheema 8 months ago
parent
commit
a766e989bb
2 changed files with 6 additions and 3 deletions
  1. 6 2
      Dockerfile
  2. 0 1
      exo/main.py

+ 6 - 2
Dockerfile

@@ -29,6 +29,10 @@ RUN pip install --no-cache-dir numpy
 # Add library path
 ENV LD_LIBRARY_PATH="/usr/local/lib:${LD_LIBRARY_PATH}"
 
-COPY test.py .
+COPY setup.py .
+COPY exo ./exo
 
-CMD ["python3", "test.py"]
+RUN sed -i '/mlx==/d' setup.py && \
+    pip install --no-cache-dir .
+
+CMD ["exo", "--inference-engine", "mlx"]

+ 0 - 1
exo/main.py

@@ -2,7 +2,6 @@ import argparse
 import asyncio
 import signal
 import json
-import logging
 import time
 import traceback
 import uuid