File size: 1,167 Bytes
cf520c9 | 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 | from setuptools import setup, find_packages
setup(
name="self-healing-training",
version="1.0.0",
description="Fully autonomous self-healing layer for Hugging Face TRL trainers",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
author="Autonomous ML Intern",
url="https://huggingface.co/ScottzillaSystems/self-healing-training",
packages=find_packages(),
python_requires=">=3.9",
install_requires=[
"torch>=2.0.0",
"transformers>=4.40.0",
"trl>=0.9.0",
],
extras_require={
"trackio": ["trackio"],
"dev": ["pytest", "pytest-cov", "black", "ruff"],
},
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
) |