Plug-in-Expert/setup.py at flexlora-dev · Heeqw/Plug-in-Expert · 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
"""
Setup script for FlexLoRA.
"""

from setuptools import setup, find_packages

setup(
    name="flexlora",
    version="0.1.0",
    packages=find_packages(),
    install_requires=[
        "torch>=2.0.0",
        "transformers>=4.36.0",
        "peft>=0.8.2",
        "bitsandbytes>=0.41.0",
        "datasets",
        "tqdm",
        "rich",
        "sentencepiece",
        "accelerate",
    ],
    author="Your Team",
    author_email="your.email@example.com",
    description="A flexible and modular QLoRA framework",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    url="https://github.com/yourusername/flexlora",
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    python_requires=">=3.8",
)