Update pyproject.toml to include transformers package and remove redundant installation step in Dockerfile.ci for cleaner CI setup.
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
1 parent
b7f297c8c3
commit
05784d3dd0
2 files changed
+2
-12
No files matched your search
@@ -34,17 +34,6 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm
|
||||
# Install vllm-omni into the same uv-managed Python environment used by the base image.
|
||||
RUN uv pip install --python "$(python3 -c 'import sys; print(sys.executable)')" --no-cache-dir ".[dev]"
|
||||
|
||||
# Debug: print key dependency versions for CI troubleshooting
|
||||
RUN python3 -c "\
|
||||
import transformers, torch, vllm; \
|
||||
print(f'transformers={transformers.__version__}'); \
|
||||
print(f'torch={torch.__version__}'); \
|
||||
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 []
|
||||
+2
-1
@@ -48,7 +48,8 @@ dev = [
|
||||
"soundfile>=0.13.1",
|
||||
"imageio[ffmpeg]>=0.6.0",
|
||||
"opencv-python>=4.12.0.88",
|
||||
"mooncake-transfer-engine==0.3.8.post1"
|
||||
"mooncake-transfer-engine==0.3.8.post1",
|
||||
"transformers==4.57.6"
|
||||
]
|
||||
|
||||
docs = [
|
||||
|
||||
Reference in new issue
Block a user