Pin transformers <4.52 and unsloth_zoo==2025.11.1 for API compat
Browse files- Dockerfile +1 -1
- requirements-training.txt +1 -1
Dockerfile
CHANGED
|
@@ -37,7 +37,7 @@ COPY --chown=user requirements-training.txt .
|
|
| 37 |
RUN pip install --no-cache-dir -r requirements-training.txt
|
| 38 |
|
| 39 |
# 4. Unsloth --no-deps (avoids torchao>=0.13 conflict)
|
| 40 |
-
RUN pip install --no-cache-dir --no-deps unsloth==2025.11.1 unsloth_zoo
|
| 41 |
|
| 42 |
# 5. Remove torchao if pulled in (incompatible with torch 2.6, crashes transformers)
|
| 43 |
RUN pip uninstall -y torchao 2>/dev/null; true
|
|
|
|
| 37 |
RUN pip install --no-cache-dir -r requirements-training.txt
|
| 38 |
|
| 39 |
# 4. Unsloth --no-deps (avoids torchao>=0.13 conflict)
|
| 40 |
+
RUN pip install --no-cache-dir --no-deps unsloth==2025.11.1 unsloth_zoo==2025.11.1
|
| 41 |
|
| 42 |
# 5. Remove torchao if pulled in (incompatible with torch 2.6, crashes transformers)
|
| 43 |
RUN pip uninstall -y torchao 2>/dev/null; true
|
requirements-training.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Training deps (torch, unsloth installed separately in Dockerfile)
|
| 2 |
-
transformers>=4.
|
| 3 |
trl>=0.12.0,<0.16
|
| 4 |
peft>=0.13.0
|
| 5 |
accelerate>=1.0.0
|
|
|
|
| 1 |
# Training deps (torch, unsloth installed separately in Dockerfile)
|
| 2 |
+
transformers>=4.46.0,<4.52
|
| 3 |
trl>=0.12.0,<0.16
|
| 4 |
peft>=0.13.0
|
| 5 |
accelerate>=1.0.0
|