New web interface, move llm engine to separate process
This commit is contained in:
33
tools/qwq_train/setup.py
Normal file
33
tools/qwq_train/setup.py
Normal file
@@ -0,0 +1,33 @@
|
||||
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',
|
||||
)
|
||||
Reference in New Issue
Block a user