petals/requirements.txt

9 lines
184 B
Plaintext
Raw Normal View History

torch>=1.12
bitsandbytes==0.33.0 #TODO update this to 0.33.2 asap
accelerate==0.10.0
huggingface-hub==0.7.0
transformers==4.21.3
protobuf>=3.12.2,<4.0.0
hivemind==1.1.1
humanfriendly