Fix Dockerfile verification: use generic transformers import check

EsmModel was reorganized in newer transformers versions.

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

View File

@@ -16,7 +16,7 @@ RUN pip install --no-cache-dir \
# Verify installations # Verify installations
RUN python -c "import torch; print(f'PyTorch: {torch.__version__}')" && \ RUN python -c "import torch; print(f'PyTorch: {torch.__version__}')" && \
python -c "from transformers import EsmModel; print('transformers: OK')" && \ python -c "import transformers; print(f'transformers: {transformers.__version__}')" && \
python -c "import chai_lab; print('chai_lab: OK')" && \ python -c "import chai_lab; print('chai_lab: OK')" && \
chai --help chai --help