K446 commited on
Commit
b724812
·
1 Parent(s): 08c4515

Pin transformers <4.52 and unsloth_zoo==2025.11.1 for API compat

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. 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.51.3,<5.0
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