Spaces:
Runtime error
Runtime error
Commit ·
21d2cbd
1
Parent(s): e82a201
Pin transformers==4.57.6 and add runtime version guard
Browse filestransformers 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>
- requirements.txt +1 -1
- src/__init__.py +10 -0
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
datasets>=2.14.0
|
| 2 |
-
transformers
|
| 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}
|