testing-fish-model / code /requirements.txt
hfastino's picture
Upload trained model to hfastino/broke-fish
b674efb verified
raw
history blame contribute delete
424 Bytes
# 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