self-hosted/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-01-02 12:48:35 +02:00
Code Issues Releases Activity

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

Browse Source 
* 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
This commit is contained in:
Skyler Mäntysaari 2023-02-23 17:54:04 +02:00 committed by GitHub
parent de5a6b2c35
commit b00631d186
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docker/docker-compose.yml
View File

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

2
machine-learning/Dockerfile
View File

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