weights/tts/__init__.py

from .configuration_nanotts import NanoTTSConfig
from .modeling_nanotts_global_local import (
    NanoTTSGenerationOutput,
    NanoTTSGlobalLocalForCausalLM,
    NanoTTSOutput,
)
from .tokenization_nanotts_sentencepiece import NanoTTSSentencePieceTokenizer

try:
    NanoTTSConfig.register_for_auto_class()
except Exception:
    pass

for auto_class_name in ("AutoModel", "AutoModelForCausalLM"):
    try:
        NanoTTSGlobalLocalForCausalLM.register_for_auto_class(auto_class_name)
    except Exception:
        pass

try:
    NanoTTSSentencePieceTokenizer.register_for_auto_class("AutoTokenizer")
except Exception:
    pass

__all__ = [
    "NanoTTSConfig",
    "NanoTTSGlobalLocalForCausalLM",
    "NanoTTSSentencePieceTokenizer",
    "NanoTTSGenerationOutput",
    "NanoTTSOutput",
]