akryldigital commited on
Commit
f17dc4f
·
verified ·
1 Parent(s): dd13e57
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  # FROM python:3.11-slim
2
  # Use HuggingFace's pre-built GPU image (includes CUDA, PyTorch, Transformers)
3
- FROM huggingface/transformers-pytorch-gpu:4.44.2
4
 
5
 
6
  WORKDIR /app
 
1
  # FROM python:3.11-slim
2
  # Use HuggingFace's pre-built GPU image (includes CUDA, PyTorch, Transformers)
3
+ FROM huggingface/transformers-pytorch-gpu:latest
4
 
5
 
6
  WORKDIR /app