gary-boon Claude Opus 4.5 commited on
Commit
1b73605
·
1 Parent(s): b788304

fix: Update torch to 2.3+ for transformers compatibility

Browse files

transformers 4.44+ requires torch.utils._pytree.register_pytree_node
which was added in torch 2.3.0. Old torch 2.1.0 only had the
underscore-prefixed _register_pytree_node.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <[email protected]>

Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -6,7 +6,8 @@ python-multipart==0.0.6
6
  pydantic==2.5.0
7
 
8
  # Machine Learning
9
- torch==2.1.0
 
10
  transformers>=4.44.0
11
  accelerate>=0.30.0
12
 
 
6
  pydantic==2.5.0
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