Skip to content

Commit ded9bc6

Browse files
authored
Fix Docker images for Liger (#4522)
1 parent fd04760 commit ded9bc6

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

docker/trl-dev/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime
1+
FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel
22
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
33
RUN pip install --upgrade pip uv
44
RUN uv pip install --system --no-cache "git+https://github.com/huggingface/trl.git#egg=trl[liger,peft,vlm]"
5-
RUN uv pip install --system hf_transfer kernels liger_kernel peft trackio
5+
RUN uv pip install --system kernels liger_kernel peft trackio

docker/trl/Dockerfile

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1-
FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime
1+
FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel
2+
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
23
RUN pip install --upgrade pip uv
3-
RUN uv pip install --system trl[liger,peft,vlm] hf_transfer kernels trackio
4+
RUN uv pip install --system trl[liger,peft,vlm] kernels trackio

0 commit comments

Comments
 (0)