diff --git a/Dockerfile b/Dockerfile index c6a8e23..da24151 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,4 +3,4 @@ FROM git.modelhub.org.cn:9443/enginex-iluvatar/bi100-3.2.3-x86-ubuntu20.04-py3.1 RUN mkdir /workspace WORKDIR /workspace/ COPY ./qwen3_6_scripts /workspace/qwen3_6_scripts -RUN cd ./qwen3_6_scripts && ./patch_ops.sh +RUN cd ./qwen3_6_scripts && ./patch_ops.sh \ No newline at end of file diff --git a/qwen3_6_scripts/patch_ops.sh b/qwen3_6_scripts/patch_ops.sh index 82d23c1..1035b2f 100755 --- a/qwen3_6_scripts/patch_ops.sh +++ b/qwen3_6_scripts/patch_ops.sh @@ -42,8 +42,7 @@ cp ./paged_attn.py /usr/local/corex/lib/python3/dist-packages/vllm/attention/ops python3 ./patch_model_runner.py # --- transformers: Qwen3_5 tokenizer / model files -------------------------- -pip install transformers==4.55.3 -i https://pypi.tuna.tsinghua.edu.cn/simple || \ - echo "[patch_ops] WARN: transformers==4.55.3 install failed; continuing with installed transformers." +pip install transformers==4.55.3 -i https://pypi.tuna.tsinghua.edu.cn/simple cp -r ./qwen3_5 /usr/local/lib/python3.10/site-packages/transformers/models/ cp -r ./qwen3_5_moe /usr/local/lib/python3.10/site-packages/transformers/models/ python3 ./patch_transformers_qwen3_5.py