Spaces:

akryldigital
/

audit_assistant

Sleeping

akryldigital commited on 23 days ago

Commit

074129d

verified ·

1 Parent(s): 0bc84fc

fix the base image version

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,7 +1,7 @@
 # FROM python:3.11-slim
 # Use HuggingFace's pre-built GPU image (includes CUDA, PyTorch, Transformers)
 # FROM huggingface/transformers-pytorch-amd-gpu:latest
-FROM huggingface/transformers-pytorch-gpu:latest
 WORKDIR /app
@@ -73,7 +73,7 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
     CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 #temp developement commands
-RUN python3 -c "import torch; print(torch.cuda.is_available(), torch.cuda.get_device_name(0))"
 # RUN mkdir /app/conversations && chmod -R 777 conversations
 # RUN mkdir /app/feedback && chmod -R 777 feedback

 # FROM python:3.11-slim
 # Use HuggingFace's pre-built GPU image (includes CUDA, PyTorch, Transformers)
 # FROM huggingface/transformers-pytorch-amd-gpu:latest
+FROM huggingface/transformers-pytorch-gpu:latest@sha256:4c7317881a534b22e18add49c925096fa902651fb0571c69f3cad58af3ea2c0f
 WORKDIR /app
     CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 #temp developement commands
+RUN python3 -c "import torch; print(torch.cuda.is_available(), torch.cuda.get_device_name(0))" || true
 # RUN mkdir /app/conversations && chmod -R 777 conversations
 # RUN mkdir /app/feedback && chmod -R 777 feedback