Pin transformers<5.0 for EsmModel compatibility

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-03-16 13:39:57 +01:00
parent 5ac0bfc25e
commit 895f3c5f75

View File

@@ -9,10 +9,10 @@ ENV DEBIAN_FRONTEND=noninteractive \
# Set working directory # Set working directory
WORKDIR /workspace WORKDIR /workspace
# Install chai_lab and transformers in a single layer # Install chai_lab and compatible transformers in a single layer
RUN pip install --no-cache-dir \ RUN pip install --no-cache-dir \
chai_lab==0.5.2 \ chai_lab==0.5.2 \
"transformers>=4.30.0" "transformers>=4.30.0,<5.0.0"
# Verify installations # Verify installations
RUN python -c "import torch; print(f'PyTorch: {torch.__version__}')" && \ RUN python -c "import torch; print(f'PyTorch: {torch.__version__}')" && \