Synchronize repository with Athena deployment
This commit is contained in:
1 parent
040a2df48b
commit
fce9900389
60 files changed
+2570
-607
No files matched your search
@@ -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"]
|
||||
Reference in new issue
Block a user