mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +00:00
fef7257fe0
The goals of these changes are: - Make Petals work in Colab right after just doing `pip install -r requirements.txt` - Make tests work independently of the protobuf package version chosen while installing dependencies
9 lines
225 B
Plaintext
9 lines
225 B
Plaintext
torch>=1.12
|
|
bitsandbytes==0.34.0
|
|
accelerate==0.10.0
|
|
huggingface-hub==0.7.0
|
|
transformers==4.21.3
|
|
protobuf>=3.20.3,<4.0dev
|
|
git+https://github.com/learning-at-home/hivemind@1e4af434f35ad43208e7e5df569c5ff5eb79681b
|
|
humanfriendly
|