Skip to content

Commit

Permalink
Fix fastapi to address CVE-2024-24762
Browse files Browse the repository at this point in the history
Signed-off-by: Heba Elayoty <hebaelayoty@gmail.com>
  • Loading branch information
helayoty committed Mar 17, 2024
1 parent c9b2bcd commit ebf2f46
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 1 deletion.
2 changes: 1 addition & 1 deletion docker/presets/llama-2/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ WORKDIR /workspace/llama
RUN ["/bin/bash", "-c", "sed -i $'/torch.distributed.init_process_group(\"nccl\")/c\\ import datetime\\\n torch.distributed.init_process_group(\"nccl\", timeout=datetime.timedelta(days=365*100))' /workspace/llama/llama/generation.py"]

RUN pip install -e .
RUN pip install torch==2.1.0 fastapi==0.103.2 pydantic==1.10.9 gputil==1.4.0
RUN pip install torch==2.1.0 fastapi==0.109.1 pydantic==1.10.9 gputil==1.4.0
RUN pip install 'uvicorn[standard]'

ARG WEIGHTS_PATH
Expand Down
14 changes: 14 additions & 0 deletions presets/tuning/tfs/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
datasets==2.16.1
peft==0.8.2
transformers==4.38.2
torch==2.2.0
accelerate==0.27.2
fastapi==0.109.1
pydantic==1.10.9
uvicorn[standard]==0.23.2
bitsandbytes==0.42.0
gputil==1.4.0
loralib
einops
xformers
deepspeed

0 comments on commit ebf2f46

Please sign in to comment.