Rebuild private TimeBraid 2.5B release with current inference runtime

0acbed8 verified about 1 month ago

2.75 kB

"""Public TimeBraid package with lazy Hugging Face runtime exports."""

__all__ = [
    # Runtime types
    "GenerationRoute",
    "TimeBraid",
    "TimeBraidBatchFeature",
    "TimeBraidConfig",
    "TimeBraidGenerateOutput",
    "TimeBraidLossBreakdown",
    "TimeBraidOutput",
    "TimeBraidProcessor",
    # Registration
    "register_timebraid_auto_classes",
    # Loading and validation
    "LoadedTimeBraid",
    "ValidatedTimeBraidCheckpointSource",
    "load_timebraid_checkpoint",
    "load_timebraid_tokenizer",
    "validate_timebraid_canonical_artifact",
    "validate_timebraid_checkpoint_source",
    "validate_timebraid_model",
]


def __getattr__(name: str):
    if name not in __all__:
        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
    from .model import (
        LoadedTimeBraid,
        TimeBraid,
        TimeBraidConfig,
        TimeBraidGenerateOutput,
        TimeBraidLossBreakdown,
        TimeBraidOutput,
        ValidatedTimeBraidCheckpointSource,
        load_timebraid_checkpoint,
        load_timebraid_tokenizer,
        validate_timebraid_canonical_artifact,
        validate_timebraid_checkpoint_source,
        validate_timebraid_model,
    )
    from .model.mot.generation_route import GenerationRoute
    from .model.timebraid import register_timebraid_auto_classes
    from .processing_timebraid import TimeBraidBatchFeature, TimeBraidProcessor

    # Importing any public name registers the Hugging Face auto classes, so
    # `AutoModelForCausalLM.from_pretrained` resolves without remote code.
    # `register_timebraid_auto_classes` is exported as well, because relying on
    # an import side effect forces callers to suppress an unused-import warning.
    register_timebraid_auto_classes()
    return {
        "GenerationRoute": GenerationRoute,
        "LoadedTimeBraid": LoadedTimeBraid,
        "TimeBraid": TimeBraid,
        "TimeBraidBatchFeature": TimeBraidBatchFeature,
        "TimeBraidConfig": TimeBraidConfig,
        "TimeBraidGenerateOutput": TimeBraidGenerateOutput,
        "TimeBraidLossBreakdown": TimeBraidLossBreakdown,
        "TimeBraidOutput": TimeBraidOutput,
        "TimeBraidProcessor": TimeBraidProcessor,
        "ValidatedTimeBraidCheckpointSource": ValidatedTimeBraidCheckpointSource,
        "load_timebraid_checkpoint": load_timebraid_checkpoint,
        "load_timebraid_tokenizer": load_timebraid_tokenizer,
        "register_timebraid_auto_classes": register_timebraid_auto_classes,
        "validate_timebraid_canonical_artifact": validate_timebraid_canonical_artifact,
        "validate_timebraid_checkpoint_source": validate_timebraid_checkpoint_source,
        "validate_timebraid_model": validate_timebraid_model,
    }[name]

	"""Public TimeBraid package with lazy Hugging Face runtime exports."""

	__all__ = [
	# Runtime types
	"GenerationRoute",
	"TimeBraid",
	"TimeBraidBatchFeature",
	"TimeBraidConfig",
	"TimeBraidGenerateOutput",
	"TimeBraidLossBreakdown",
	"TimeBraidOutput",
	"TimeBraidProcessor",
	# Registration
	"register_timebraid_auto_classes",
	# Loading and validation
	"LoadedTimeBraid",
	"ValidatedTimeBraidCheckpointSource",
	"load_timebraid_checkpoint",
	"load_timebraid_tokenizer",
	"validate_timebraid_canonical_artifact",
	"validate_timebraid_checkpoint_source",
	"validate_timebraid_model",
	]


	def __getattr__(name: str):
	if name not in __all__:
	raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
	from .model import (
	LoadedTimeBraid,
	TimeBraid,
	TimeBraidConfig,
	TimeBraidGenerateOutput,
	TimeBraidLossBreakdown,
	TimeBraidOutput,
	ValidatedTimeBraidCheckpointSource,
	load_timebraid_checkpoint,
	load_timebraid_tokenizer,
	validate_timebraid_canonical_artifact,
	validate_timebraid_checkpoint_source,
	validate_timebraid_model,
	)
	from .model.mot.generation_route import GenerationRoute
	from .model.timebraid import register_timebraid_auto_classes
	from .processing_timebraid import TimeBraidBatchFeature, TimeBraidProcessor

	# Importing any public name registers the Hugging Face auto classes, so
	# `AutoModelForCausalLM.from_pretrained` resolves without remote code.
	# `register_timebraid_auto_classes` is exported as well, because relying on
	# an import side effect forces callers to suppress an unused-import warning.
	register_timebraid_auto_classes()
	return {
	"GenerationRoute": GenerationRoute,
	"LoadedTimeBraid": LoadedTimeBraid,
	"TimeBraid": TimeBraid,
	"TimeBraidBatchFeature": TimeBraidBatchFeature,
	"TimeBraidConfig": TimeBraidConfig,
	"TimeBraidGenerateOutput": TimeBraidGenerateOutput,
	"TimeBraidLossBreakdown": TimeBraidLossBreakdown,
	"TimeBraidOutput": TimeBraidOutput,
	"TimeBraidProcessor": TimeBraidProcessor,
	"ValidatedTimeBraidCheckpointSource": ValidatedTimeBraidCheckpointSource,
	"load_timebraid_checkpoint": load_timebraid_checkpoint,
	"load_timebraid_tokenizer": load_timebraid_tokenizer,
	"register_timebraid_auto_classes": register_timebraid_auto_classes,
	"validate_timebraid_canonical_artifact": validate_timebraid_canonical_artifact,
	"validate_timebraid_checkpoint_source": validate_timebraid_checkpoint_source,
	"validate_timebraid_model": validate_timebraid_model,
	}[name]