diff --git a/requirements-dev.txt b/requirements-dev.txt new file mode 100644 index 0000000..637434d --- /dev/null +++ b/requirements-dev.txt @@ -0,0 +1,6 @@ +pytest==6.2.5 # see https://github.com/pytest-dev/pytest/issues/9621 +pytest-forked +pytest-asyncio==0.16.0 +black==22.3.0 +isort==5.10.1 +psutil \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..3a495bf --- /dev/null +++ b/requirements.txt @@ -0,0 +1,9 @@ +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