setup.py

from setuptools import setup

setup(name='allamo',
      version='5.0.0',
      author='Krzysztof (Chris) Ociepa',
      packages=['allamo'],
      description='Simple, hackable and fast implementation for training/finetuning medium-sized LLaMA-based models',
      license='MIT',
      install_requires=[
            'torch',
            'numpy',
            'joblib',
            'wandb'
      ],
)