BloomBee/setup.cfg at main · ai-decentralized/BloomBee · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
[metadata]
name = bloombee
version = attr: bloombee.__version__
author = Sophie
author_email = syang127@ucmerced.edu
description = A short description of your project
long_description = file: README.md
long_description_content_type = text/markdown
url = https://github.com/yottalabsai/bloombee
project_urls =
    Bug Tracker = https://github.com/yottalabsai/bloombee/issues
classifiers =
    Development Status :: 4 - Beta
    Intended Audience :: Developers
    Intended Audience :: Science/Research
    License :: OSI Approved :: Apache Software License
    Programming Language :: Python :: 3
    Programming Language :: Python :: 3.8
    Programming Language :: Python :: 3.9
    Programming Language :: Python :: 3.10
    Programming Language :: Python :: 3.11
    Topic :: Scientific/Engineering
    Topic :: Scientific/Engineering :: Mathematics
    Topic :: Scientific/Engineering :: Artificial Intelligence
    Topic :: Software Development
    Topic :: Software Development :: Libraries
    Topic :: Software Development :: Libraries :: Python Modules

[options]
package_dir =
    = src
packages = find:
python_requires = >=3.9
install_requires =
    torch>=1.12
    bitsandbytes==0.41.0
    accelerate>=0.27.2
    huggingface-hub>=0.11.1,<1.0.0
    tokenizers>=0.13.3
    transformers>=5.5.0  # if you change this, please also change version assert in bloombee/__init__.py
    speedtest-cli==2.1.3
    hivemind @ git+https://github.com/learning-at-home/hivemind.git@4bd43b77895019b20d18d81d0d0c1a5ab9a10847
    PyYAML
    scipy>=1.2.1
    prefetch_generator>=1.0.1
    msgpack>=0.5.6
    sortedcontainers
    uvloop>=0.14.0
    protobuf>=5.29.0
    configargparse>=1.2.3
    py-multihash>=0.2.3
    cryptography>=3.4.6
    pydantic>=2.0.0
    varint>=1.0.2
    base58>=1.0.2
    netaddr>=1.3.0
    idna>=3.10
    py-cid>=0.3.0
    tensor_parallel==1.0.23
    humanfriendly
    async-timeout>=4.0.2
    cpufeature>=0.2.0; platform_machine == "x86_64"
    packaging>=20.9
    sentencepiece>=0.1.99
    peft==0.8.2
    safetensors>=0.3.1
    Dijkstar>=2.6.0
    numpy<2

[options.extras_require]
dev =
    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