File size: 359 Bytes
2ef4ea4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# Core
torch>=2.4
transformers>=5.0
datasets>=3.0
accelerate>=1.0
tokenizers>=0.20

# GraphCodeBERT data-flow extraction (tree-sitter is required by the original
# GraphCodeBERT preprocessing pipeline -- it is not an optional extra).
tree-sitter>=0.21
tree-sitter-python>=0.21

# Data / metrics
numpy>=1.24
pyarrow>=14
scikit-learn>=1.3
huggingface-hub>=0.26