H-Liu1997 commited on
Commit
975c258
·
1 Parent(s): da265c6

fix: pin transformers<5 for google/umt5-base compat (no model_type in config)

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -7,7 +7,7 @@ RUN apt-get update && apt-get install -y \
7
  RUN pip3 install --no-cache-dir \
8
  torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
9
  pip3 install --no-cache-dir \
10
- flask flask-cors numpy transformers accelerate sentencepiece protobuf \
11
  lightning omegaconf safetensors diffusers ftfy regex
12
 
13
  # Create non-root user (HF Spaces runs as uid 1000)
 
7
  RUN pip3 install --no-cache-dir \
8
  torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
9
  pip3 install --no-cache-dir \
10
+ flask flask-cors numpy "transformers>=4.40,<5" accelerate sentencepiece protobuf \
11
  lightning omegaconf safetensors diffusers ftfy regex
12
 
13
  # Create non-root user (HF Spaces runs as uid 1000)