"megatron/inference/tokenization.py" did not exist on "554d1cc04242bc2273630c672af5ac5f6e2883fe"
requirements_flax.txt 83 Bytes
Newer Older
YiYi Xu's avatar
YiYi Xu committed
1
2
3
4
5
6
7
8
9
transformers>=4.25.1
datasets
flax
optax
torch
torchvision
ftfy
tensorboard
Jinja2