diff --git a/.gitignore b/.gitignore index ede8ee2..5a5d090 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,6 @@ *.egg-info .vscode/* */**/.DS_Store -**/__pycache__/ \ No newline at end of file +**/__pycache__/ + +build/ diff --git a/fireredtts2/__init__.py b/fireredtts2/__init__.py index e69de29..8c0401a 100644 --- a/fireredtts2/__init__.py +++ b/fireredtts2/__init__.py @@ -0,0 +1,4 @@ + +from .fireredtts2 import FireRedTTS2 + +__all__ = ["FireRedTTS2"] diff --git a/fireredtts2/utils/__init__.py b/fireredtts2/utils/__init__.py new file mode 100644 index 0000000..c1decd2 --- /dev/null +++ b/fireredtts2/utils/__init__.py @@ -0,0 +1,20 @@ +"""Utilities package for fireredtts2. + +This module re-exports the commonly used helpers from the internal +`spliter` module so imports like `from fireredtts2.utils.spliter import ...` +continue to work when the package is installed. +""" + +from .spliter import ( + clean_text, + split_text, + process_text, + process_text_list, +) + +__all__ = [ + "clean_text", + "split_text", + "process_text", + "process_text_list", +] diff --git a/setup.py b/setup.py index 158e2b4..0ce4795 100644 --- a/setup.py +++ b/setup.py @@ -1,3 +1,25 @@ +from pathlib import Path from setuptools import setup, find_packages -setup(name="fireredtts2", version="0.1", packages=find_packages()) + +HERE = Path(__file__).parent +long_description = (HERE / "README.md").read_text() + +setup( + name="fireredtts2", + version="0.1.0", + description="FireRedTTS2 - speech generation utilities and model wrapper", + long_description=long_description, + long_description_content_type="text/markdown", + author="FireRedTTS2", + packages=find_packages(exclude=("tests", "examples")), + include_package_data=True, + install_requires=[ + "torchtune", + "torchao", + "transformers", + "einops", + "gradio", + ], + python_requires=">=3.8", +)