File size: 424 Bytes
b674efb
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
# Requirements for GLiNER2 Multi-Model Endpoint
# NOTE: These are installed when the SageMaker container starts
#
# IMPORTANT: SageMaker PyTorch 2.1.0 container requires transformers<4.46
# (transformers 4.46+ uses torch.utils._pytree.register_pytree_node which needs PyTorch 2.3+)

# Core dependencies - gliner2 must be installed for model loading
gliner2==1.0.1
transformers>=4.30.0,<4.46.0

# JSON handling
orjson>=3.9.0