Siddh12334 commited on
Commit
d1de63d
·
verified ·
1 Parent(s): b696295

chore: update comment on torchao shim strategy

Browse files
Files changed (1) hide show
  1. 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 — register_constant shim applied at runtime for torchao compat
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 \