This commit is contained in:
parent
524d2f84f8
commit
f15cdf981b
|
|
@ -99,7 +99,7 @@ RUN mkdir -p /wheels && \
|
||||||
# ── ✅ 再打包 runtime 阶段必需依赖 ────────────────────────────────────────────
|
# ── ✅ 再打包 runtime 阶段必需依赖 ────────────────────────────────────────────
|
||||||
RUN pip wheel \
|
RUN pip wheel \
|
||||||
pydantic orjson psutil pyzmq pynvml \
|
pydantic orjson psutil pyzmq pynvml \
|
||||||
transformers==4.52.0 uvicorn fastapi IPython aiohttp \
|
transformers==4.56.0 uvicorn fastapi IPython aiohttp \
|
||||||
setproctitle uvloop sentencepiece triton pillow cachetools msgspec blake3 cloudpickle compressed-tensors einops openai py-cpuinfo dill partial_json_parser python-multipart torchao \
|
setproctitle uvloop sentencepiece triton pillow cachetools msgspec blake3 cloudpickle compressed-tensors einops openai py-cpuinfo dill partial_json_parser python-multipart torchao \
|
||||||
-w /wheels
|
-w /wheels
|
||||||
|
|
||||||
|
|
@ -137,7 +137,7 @@ COPY --from=builder-extras /wheels /tmp/wheels
|
||||||
# ✅ 优先装你自编的 torch,避免被 PyPI 上的覆盖
|
# ✅ 优先装你自编的 torch,避免被 PyPI 上的覆盖
|
||||||
RUN ls -lh /tmp/wheels && \
|
RUN ls -lh /tmp/wheels && \
|
||||||
rm -f /tmp/wheels/torch-2.7.1a0+*.whl && \
|
rm -f /tmp/wheels/torch-2.7.1a0+*.whl && \
|
||||||
rm -f /tmp/wheels/huggingface_hub-0.33.4*.whl && \
|
rm -f /tmp/wheels/huggingface_hub-0.34.4*.whl && \
|
||||||
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/torch*.whl && \
|
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/torch*.whl && \
|
||||||
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/vllm-*.whl && \
|
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/vllm-*.whl && \
|
||||||
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/sgl_kernel-*.whl && \
|
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/sgl_kernel-*.whl && \
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue