chore: update comment on torchao shim strategy
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -4,7 +4,7 @@ WORKDIR /app
|
|
| 4 |
|
| 5 |
RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
|
| 6 |
|
| 7 |
-
# Stable cu121 torch —
|
| 8 |
RUN pip install --no-cache-dir \
|
| 9 |
torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
|
| 10 |
pip install --no-cache-dir \
|
|
|
|
| 4 |
|
| 5 |
RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
|
| 6 |
|
| 7 |
+
# Stable cu121 torch — TorchAO import shims are applied at runtime before Unsloth loads.
|
| 8 |
RUN pip install --no-cache-dir \
|
| 9 |
torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
|
| 10 |
pip install --no-cache-dir \
|