34 lines
737 B
Python
34 lines
737 B
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
name="train",
|
|
version="0.1.0",
|
|
packages=find_packages(),
|
|
scripts=[
|
|
'bin/train'
|
|
],
|
|
|
|
install_requires=[
|
|
'accelerate>=0.26.0',
|
|
'bitsandbytes>=0.45.0',
|
|
'black>=22.0.0',
|
|
'datasets>=2.14.6',
|
|
'einops>=0.7.0',
|
|
'flake8>=4.0.0',
|
|
'ipykernel>=6.0.0',
|
|
'ipywidgets>=8.0.0',
|
|
'peft>=0.8.0',
|
|
'pytest-cov>=4.0.0',
|
|
'pytest>=7.0.0',
|
|
'pyyaml>=6.0',
|
|
'requests>=2.28.0',
|
|
'sentencepiece>=0.1.99',
|
|
'torch>=2.0.0',
|
|
'transformers>=4.30.0',
|
|
'trl>=0.7.8',
|
|
'unsloth>=2025.3',
|
|
'vllm==0.8.2',
|
|
],
|
|
python_requires='>=3.10',
|
|
)
|