Synchronize repository with Athena deployment

This commit is contained in:
Mikei386 committed 2026-09-13 20:01:36 +02:00
1 parent 040a2df48b
commit fce9900389
60 files changed
+2570 -607

No files matched your search

+3 -1
View File
@@ -13,9 +13,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends python3.12-venv
"transformers==${TRANSFORMERS_VERSION}" \
"accelerate==${ACCELERATE_VERSION}" \
"huggingface-hub==${HF_HUB_VERSION}" \
"nvidia-modelopt==0.46.0" bitsandbytes \
sentencepiece protobuf safetensors pillow && \
useradd --system --uid 10002 --home /nonexistent --shell /usr/sbin/nologin image-worker
COPY image_worker.py /app/image_worker.py
COPY image_worker_9b.py /app/image_worker_9b.py
USER 10002:10002
ENTRYPOINT ["/opt/image-venv/bin/python", "/app/image_worker.py"]
ENTRYPOINT ["/opt/image-venv/bin/python", "/app/image_worker_9b.py"]