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