From c3d256271e0f7a5c30bb16fadc98232dfd510f09 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Fri, 26 May 2023 10:37:02 -0400 Subject: [PATCH] fix wheel install glob --- docker/Dockerfile-base | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base index 868e0a08d..d2abd9bca 100644 --- a/docker/Dockerfile-base +++ b/docker/Dockerfile-base @@ -74,7 +74,7 @@ COPY --from=flash-attn-builder /workspace/flash-attention/csrc/xentropy/dist/xen COPY --from=flash-attn-builder /workspace/flash-attention/csrc/rotary/dist/rotary_emb-*.whl wheels COPY --from=flash-attn-builder /workspace/flash-attention/csrc/layer_norm/dist/dropout_layer_norm-*.whl wheels -RUN pip3 install wheels/deepspeed-*.whl wheels/flash_attn-*.whl wheels/fused_dense_lib-*.whl wheels/xentropy-*.whl wheels/rotary-*.whl wheels/dropout_layer_norm-*.whl +RUN pip3 install wheels/deepspeed-*.whl wheels/flash_attn-*.whl wheels/fused_dense_lib-*.whl wheels/xentropy_cuda_lib-*.whl wheels/rotary_emb-*.whl wheels/dropout_layer_norm-*.whl RUN git lfs install --skip-repo RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main" \ "accelerate @ git+https://github.com/huggingface/accelerate.git@main" \