__init__.py

"""
DreamMachine HuggingFace Model Package

Copyright (c) 2026 Fangjun Wen / DreamMachine Research Team

Registers DreamMachine with HuggingFace AutoClasses so that:

    AutoConfig.from_pretrained("fangjunwen/dreammachine", trust_remote_code=True)
    AutoModel.from_pretrained("fangjunwen/dreammachine",  trust_remote_code=True)
    AutoTokenizer.from_pretrained("...", trust_remote_code=True)
    AutoFeatureExtractor.from_pretrained("...", trust_remote_code=True)

all work out of the box.
"""

from configuration_dreammachine import DreamMachineConfig
from modeling_dreammachine import DreamMachineModel, DreamMachineOutput
from tokenization_dreammachine import DreamMachineTokenizer
from feature_extraction_dreammachine import DreamMachineFeatureExtractor

try:
    from transformers import (
        AutoConfig, AutoModel, AutoTokenizer, AutoFeatureExtractor,
        AutoModelForSequenceClassification,
    )
    AutoConfig.register("dreammachine", DreamMachineConfig)
    AutoModel.register(DreamMachineConfig, DreamMachineModel)
    AutoModelForSequenceClassification.register(DreamMachineConfig, DreamMachineModel)
    AutoTokenizer.register(DreamMachineConfig, slow_tokenizer_class=DreamMachineTokenizer)
    AutoFeatureExtractor.register(DreamMachineConfig, DreamMachineFeatureExtractor)
except Exception:
    pass

__version__ = "1.0.0"
__all__ = [
    "DreamMachineConfig",
    "DreamMachineModel",
    "DreamMachineOutput",
    "DreamMachineTokenizer",
    "DreamMachineFeatureExtractor",
]