setup.py

#!/usr/bin/env python

from setuptools import setup, find_packages

setup(name='reasoners',
      version='0.0.0',
      packages=find_packages(exclude=('examples', 'examples.*')),
      entry_points={
          'console_scripts': ['reasoners-visualizer=reasoners.visualization:main'],
      },
      install_requires=['tqdm',
                        'fire',
                        'numpy',
                        'scipy',
                        'torch',
                        'datasets',
                        'transformers',
                        'sentencepiece',
                        'openai',
                        'llama1@git+https://github.com/AegeanYan/llama@llama_v1',#llama 2 have some alias problem so you may need to clone the forked llama1 at private repo. please check the setup.py
                        'llama@git+https://github.com/facebookresearch/llama@main',
                        'fairscale'])