28 lines
800 B
Python
28 lines
800 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""
|
|
HYWorld (HY-WorldPlay) model components for FastVideo.
|
|
|
|
This module provides:
|
|
- HYWorldTransformer3DModel: The main transformer model with ProPE and action conditioning
|
|
- HYWorldVideoGenerator: Extended VideoGenerator for HYWorld inference
|
|
- Utilities for pose processing and camera trajectory generation
|
|
"""
|
|
|
|
from .hyworld import HYWorldTransformer3DModel, HYWorldDoubleStreamBlock
|
|
|
|
# Inference utilities (used by examples)
|
|
from .resolution_utils import (
|
|
get_resolution_from_image,
|
|
)
|
|
|
|
__all__ = [
|
|
# Model (used by model registry)
|
|
"HYWorldTransformer3DModel",
|
|
"HYWorldDoubleStreamBlock",
|
|
# Inference utilities (used by examples)
|
|
"get_resolution_from_image",
|
|
]
|
|
|
|
# Entry point for model registry
|
|
EntryClass = HYWorldTransformer3DModel
|