|
|
|
@ -29,7 +29,7 @@ classifiers =
|
|
|
|
|
package_dir =
|
|
|
|
|
= src
|
|
|
|
|
packages = find:
|
|
|
|
|
python_requires = >=3.8,<3.11
|
|
|
|
|
python_requires = >=3.8
|
|
|
|
|
install_requires =
|
|
|
|
|
torch>=1.12
|
|
|
|
|
bitsandbytes==0.40.1.post1
|
|
|
|
@ -39,7 +39,7 @@ install_requires =
|
|
|
|
|
transformers>=4.31.0,<5.0.0
|
|
|
|
|
speedtest-cli==2.1.3
|
|
|
|
|
pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind==1.1.8
|
|
|
|
|
hivemind==1.1.8
|
|
|
|
|
hivemind @ git+https://github.com/learning-at-home/hivemind
|
|
|
|
|
tensor_parallel==1.0.23
|
|
|
|
|
humanfriendly
|
|
|
|
|
async-timeout>=4.0.2
|
|
|
|
|