Add transformers package installation in Dockerfile.ci to ensure compatibility with the CI environment.
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
379f61d038
commit
b7f297c8c3
1 file changed
+2
@@ -43,6 +43,8 @@ print(f'vllm={vllm.__version__}'); \
|
||||
print(f'cuda={torch.version.cuda}'); \
|
||||
"
|
||||
|
||||
RUN uv pip install transformers==4.57.6
|
||||
|
||||
RUN ln -sf /usr/bin/python3 /usr/bin/python
|
||||
|
||||
ENTRYPOINT []
|
||||
Reference in new issue
Block a user