vasiuuu commited on
Commit
b9ea90c
·
1 Parent(s): 4e35ea8

Fix torch.int1 error by downgrading torchao

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -11,6 +11,7 @@ WORKDIR /app
11
 
12
  # Upgrade pip and install unsloth properly
13
  RUN pip install --no-cache-dir --upgrade pip
 
14
  RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
15
 
16
  COPY --chown=user ./requirements.txt requirements.txt
 
11
 
12
  # Upgrade pip and install unsloth properly
13
  RUN pip install --no-cache-dir --upgrade pip
14
+ RUN pip install --no-cache-dir "torchao<=0.7.0"
15
  RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
16
 
17
  COPY --chown=user ./requirements.txt requirements.txt