浏览代码

fix(machine-learning): Add the command to execute at startup (#1843)

* fix(machine-learning): Add the command to execute at startup

Previously it wasn't set in the Docker container but it should be.

* fix(docker): remove machine-learning command arg

* fix(docker): machine-learning CMD argument
Skyler Mäntysaari 2 年之前
父节点
当前提交
b00631d186
共有 2 个文件被更改,包括 2 次插入1 次删除
  1. 0 1
      docker/docker-compose.yml
  2. 2 0
      machine-learning/Dockerfile

+ 0 - 1
docker/docker-compose.yml

@@ -34,7 +34,6 @@ services:
   immich-machine-learning:
     container_name: immich_machine_learning
     image: altran1502/immich-machine-learning:release
-    command: [ "python", "src/main.py" ]
     volumes:
       - ${UPLOAD_LOCATION}:/usr/src/app/upload
       - model-cache:/cache

+ 2 - 0
machine-learning/Dockerfile

@@ -14,3 +14,5 @@ RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece fl
 RUN pip install --no-deps sentence-transformers
 
 COPY . .
+
+CMD ["python", "src/main.py"]