File size: 206 Bytes
1d29f6c
 
 
 
9d4c04e
1d29f6c
 
9d4c04e
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
# Core inference (PyTorch)
transformers>=4.40
torch>=2.0

# ONNX path: pip install onnxruntime
# onnxruntime>=1.17
# numpy>=1.24

# Ensemble (ensemble/predict_ensemble.py)
# scikit-learn>=1.3
# joblib>=1.3