"official/modeling/fast_training/progressive/trainer.py" did not exist on "98839bd28c9fbf98105e45ead5c88f6f9c255c02"
requirements.txt 345 Bytes
Newer Older
1
2
ninja  # For faster builds.
psutil
3
ray >= 2.5.1
4
pandas  # Required for Ray data.
5
pyarrow  # Required for Ray data.
6
7
sentencepiece  # Required for LLaMA tokenizer.
numpy
yanxiyue's avatar
yanxiyue committed
8
torch == 2.0.1
9
10
transformers >= 4.34.0  # Required for Mistral.
xformers == 0.0.22  # Required for Mistral.
11
fastapi
12
uvicorn[standard]
Keming's avatar
Keming committed
13
pydantic < 2  # Required for OpenAI server.