chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,9 @@
|
||||
# Dockerfile used to create the docker image for `03_serving_stable_diffusion`.
|
||||
FROM anyscale/ray:latest-py39-cu118
|
||||
|
||||
COPY requirements.txt ./
|
||||
|
||||
RUN pip install --no-cache-dir -U -r requirements.txt
|
||||
|
||||
RUN echo "Testing Ray Import..." && python -c "import ray"
|
||||
RUN ray --version
|
||||
@@ -0,0 +1,8 @@
|
||||
accelerate==0.20.3
|
||||
diffusers==0.17.1
|
||||
fastapi==0.97.0
|
||||
ipywidgets
|
||||
matplotlib==3.7.1
|
||||
numpy==1.24.3
|
||||
torch==2.0.1
|
||||
transformers==4.30.1
|
||||
@@ -0,0 +1,17 @@
|
||||
# Dockerfile used to create the docker image for `04_finetuning_llms_with_deepspeed`.
|
||||
FROM anyscale/ray:2.9.0-py310-cu121
|
||||
|
||||
COPY requirements.txt ./
|
||||
|
||||
RUN sudo apt-get update
|
||||
RUN sudo apt-get install -y libaio1
|
||||
|
||||
RUN pip install --upgrade pip
|
||||
# We need pydantic at this version to install deepspeed 0.10.2 (as part of the requirements.txt)
|
||||
RUN pip install pydantic==1.10.7
|
||||
RUN pip install -U -r requirements.txt
|
||||
RUN pip install torch==2.1.1 --index-url https://download.pytorch.org/whl/cu121
|
||||
RUN pip install flash-attn==2.4.2 --no-build-isolation
|
||||
|
||||
RUN echo "Testing Ray Import..." && python -c "import ray"
|
||||
RUN ray --version
|
||||
+12
@@ -0,0 +1,12 @@
|
||||
deepspeed==0.10.2
|
||||
fairscale
|
||||
transformers>=4.36.2
|
||||
dataset
|
||||
accelerate
|
||||
evaluate
|
||||
wandb
|
||||
pytorch-lightning
|
||||
protobuf
|
||||
torchmetrics
|
||||
sentencepiece
|
||||
peft==0.7.0
|
||||
Reference in New Issue
Block a user