Replaced deepseek with qwq
This commit is contained in:
@@ -5,26 +5,27 @@ setup(
|
||||
version="0.1.0",
|
||||
packages=find_packages(),
|
||||
scripts=[
|
||||
'bin/train_deepseek',
|
||||
'bin/train_mistral'
|
||||
'bin/train'
|
||||
],
|
||||
|
||||
install_requires=[
|
||||
'pyyaml>=6.0',
|
||||
'requests>=2.28.0',
|
||||
'torch>=2.0.0',
|
||||
'transformers>=4.30.0',
|
||||
'accelerate>=0.25.0',
|
||||
'bitsandbytes>=0.41.1',
|
||||
'einops>=0.7.0',
|
||||
'sentencepiece>=0.1.99',
|
||||
'unsloth>=2025.2',
|
||||
'trl>=0.7.8',
|
||||
'datasets>=2.14.6',
|
||||
'peft>=0.8.0',
|
||||
'pytest>=7.0.0',
|
||||
'pytest-cov>=4.0.0',
|
||||
'black>=22.0.0',
|
||||
'flake8>=4.0.0'
|
||||
'datasets>=2.14.6',
|
||||
'einops>=0.7.0',
|
||||
'flake8>=4.0.0',
|
||||
'peft>=0.8.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.2',
|
||||
],
|
||||
classifiers=[
|
||||
'Development Status :: 3 - Alpha',
|
||||
|
||||
Reference in New Issue
Block a user