nicopbeard Claude Sonnet 4.6 commited on
Commit
21d2cbd
·
1 Parent(s): e82a201

Pin transformers==4.57.6 and add runtime version guard

Browse files

transformers 5.x silently produces wrong predictions (all 'unknown') when
loading a 4.x checkpoint — no error is raised. Pinning the exact version
the model was saved with, and adding a startup check, makes the failure
loud and clear instead of invisible.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Files changed (2) hide show
  1. requirements.txt +1 -1
  2. src/__init__.py +10 -0
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  datasets>=2.14.0
2
- transformers>=4.35.0,<5.0.0
3
  torch>=2.0.0
4
  pandas>=2.0.0
5
  numpy>=1.24.0
 
1
  datasets>=2.14.0
2
+ transformers==4.57.6
3
  torch>=2.0.0
4
  pandas>=2.0.0
5
  numpy>=1.24.0
src/__init__.py CHANGED
@@ -1,5 +1,15 @@
1
  from __future__ import annotations
2
 
 
 
 
 
 
 
 
 
 
 
3
  from data.schema import Argument, Debate
4
 
5
  LABEL2ID = {"claim": 0, "counter_claim": 1, "premise": 2, "unknown": 3}
 
1
  from __future__ import annotations
2
 
3
+ import importlib.metadata
4
+ from packaging.version import Version
5
+
6
+ _tv = Version(importlib.metadata.version("transformers"))
7
+ if _tv >= Version("5.0.0"):
8
+ raise RuntimeError(
9
+ f"transformers {_tv} is installed but this project requires <5.0.0. "
10
+ "Run: pip install transformers==4.57.6"
11
+ )
12
+
13
  from data.schema import Argument, Debate
14
 
15
  LABEL2ID = {"claim": 0, "counter_claim": 1, "premise": 2, "unknown": 3}