ellmer/setup.py at main · tteofili/ellmer · 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
import setuptools

with open("README.md", "r") as fh:
    long_description = fh.read()

setuptools.setup(
    name="ellmer",
    version="0.0.1",
    description="Explain Large Language Models for Entity Resolution",
    long_description=long_description,
    long_description_content_type="text/markdown",
    packages=['ellmer', 'ellmer.prompts', 'ellmer.post_hoc'],
    install_requires=[
          'pandas',
          'numpy',
          'certa',
          'langchain',
          'llama-cpp-python',
          'bitsandbytes',
          'transformers',
          'peft',
          'accelerate',
          'einops',
          'safetensors',
          'torch',
          'openai',
          'nltk',
          'langchain',
          'langchain_huggingface'
      ],
    classifiers=[
        "Programming Language :: Python :: 3",
        'License :: OSI Approved :: Apache Software License',
        "Operating System :: OS Independent",
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
    ],
    python_requires='>=3.6',
)