@@ -19,11 +19,24 @@ package_dir =
packages = find:
python_requires = >=3.7
install_requires =
- file: requirements.txt
+ 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@be88b4280cdd87432168e1da238e532f1364078b
+ humanfriendly
+ async-timeout>=4.0.2
[options.extras_require]
dev =
- file: requirements-dev.txt
+ pytest==6.2.5
+ pytest-forked
+ pytest-asyncio==0.16.0
+ black==22.3.0
+ isort==5.10.1
+ psutil
[options.packages.find]
where = src