From a191b6e192b5023813aa51574f3518c915a944d4 Mon Sep 17 00:00:00 2001 From: Olamide Isreal Date: Mon, 16 Mar 2026 17:54:19 +0100 Subject: [PATCH] Pin transformers<4.45 to fix EsmModel import at runtime Chai-1 requires EsmModel from transformers top-level import. Versions 4.45+ moved it, causing ImportError at runtime. Co-Authored-By: Claude Opus 4.6 (1M context) --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 3c8d170..b69e7d3 100755 --- a/Dockerfile +++ b/Dockerfile @@ -15,7 +15,7 @@ WORKDIR /workspace # Install chai_lab and compatible transformers in a single layer RUN pip install --no-cache-dir \ chai_lab==0.5.2 \ - "transformers>=4.30.0,<5.0.0" + "transformers>=4.30.0,<4.45.0" # Make package dirs writable for any user (K8s may run as non-root) RUN chmod -R a+rw /opt/conda/lib/python3.11/site-packages/chai_lab/ && \ @@ -26,7 +26,7 @@ RUN chmod -R a+rw /opt/conda/lib/python3.11/site-packages/chai_lab/ && \ # Verify installations RUN python -c "import torch; print(f'PyTorch: {torch.__version__}')" && \ - python -c "import transformers; print(f'transformers: {transformers.__version__}')" && \ + python -c "from transformers import EsmModel; print('EsmModel: OK')" && \ python -c "import chai_lab; print('chai_lab: OK')" && \ chai --help