| import os |
|
|
| from setuptools import setup, find_packages |
|
|
|
|
| _PATH_ROOT = os.path.dirname(__file__) |
|
|
| with open(os.path.join(_PATH_ROOT, "README.md"), encoding="utf-8") as fo: |
| readme = fo.read() |
|
|
| setup( |
| name='lit-llama', |
| version='0.1.0', |
| description='Implementation of the LLaMA language model', |
| author='Lightning AI', |
| url='https://github.com/lightning-AI/lit-llama', |
| install_requires=[ |
| "torch>=2.0.0", |
| "lightning @ git+https://github.com/Lightning-AI/lightning@master", |
| "sentencepiece", |
| "bitsandbytes", |
| ], |
| packages=find_packages(), |
| long_description=readme, |
| long_description_content_type="text/markdown", |
| ) |
|
|
