gary-boon commited on
Commit
bfdde66
·
2 Parent(s): 10baadf8b9ad4b

Merge branch 'main' into feature/score-all-classification

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -7,7 +7,8 @@ pydantic>=2.0.0 # Relaxed for mistral-common compatibility
7
 
8
  # Machine Learning
9
  # torch 2.3+ required for transformers 4.44+ (pytree API compatibility)
10
- torch>=2.3.0
 
11
  transformers>=4.44.0
12
  accelerate>=0.30.0
13
  mistral-common>=1.5.0 # Required for Devstral Tekken tokenizer
 
7
 
8
  # Machine Learning
9
  # torch 2.3+ required for transformers 4.44+ (pytree API compatibility)
10
+ # Pinned <2.11: PyTorch 2.11.0 requires CUDA drivers newer than HF Spaces GPU provides
11
+ torch>=2.3.0,<2.11.0
12
  transformers>=4.44.0
13
  accelerate>=0.30.0
14
  mistral-common>=1.5.0 # Required for Devstral Tekken tokenizer