Skip to content

Commit

Permalink
[docker] Update deepspeed docker to nightly wheel (deepjavalibrary#915)
Browse files Browse the repository at this point in the history
  • Loading branch information
tosterberg authored and KexinFeng committed Aug 16, 2023
1 parent 0986a16 commit ab2cf4f
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions serving/docker/deepspeed.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,12 @@ ARG djl_version=0.23.0~SNAPSHOT
ARG python_version=3.9
ARG torch_version=2.0.1
ARG torch_vision_version=0.15.2
ARG deepspeed_wheel="https://publish.djl.ai/deepspeed/deepspeed-0.9.2-py2.py3-none-any.whl"
ARG deepspeed_wheel="https://publish.djl.ai/deepspeed/deepspeed-nightly-py2.py3-none-any.whl"
ARG lmi_dist_wheel="https://publish.djl.ai/lmi_dist/lmi_dist-nightly-py3-none-any.whl"
ARG protobuf_version=3.20.3
ARG transformers_version=4.29.2
ARG accelerate_version=0.19.0
ARG diffusers_version=0.15.0
ARG transformers_version=4.30.2
ARG accelerate_version=0.20.3
ARG diffusers_version=0.16.0
ARG bitsandbytes_version=0.39.1
ARG peft_version=0.3.0

Expand Down

0 comments on commit ab2cf4f

Please sign in to comment.