| from .arch import get_arch_memory, is_hopper_arch | |
| from .cpu_offload_wrapper import CPUOffloadWrapper | |
| from .sequence_schema import Modality, VarlenHandler | |
| class EngineConfig: | |
| pass | |
| class DataProxyConfig: | |
| def __init__( | |
| self, | |
| t_patch_size=1, | |
| patch_size=2, | |
| frame_receptive_field=-1, | |
| spatial_rope_interpolation="extra", | |
| ref_audio_offset=1000, | |
| text_offset=0, | |
| coords_style="v2", | |
| ): | |
| self.t_patch_size = t_patch_size | |
| self.patch_size = patch_size | |
| self.frame_receptive_field = frame_receptive_field | |
| self.spatial_rope_interpolation = spatial_rope_interpolation | |
| self.ref_audio_offset = ref_audio_offset | |
| self.text_offset = text_offset | |
| self.coords_style = coords_style | |
| class EvaluationConfig: | |
| pass | |
| def parse_config(*args, **kwargs): | |
| raise RuntimeError("The upstream Magi Human CLI config parser is not used in WanGP.") | |
| __all__ = [ | |
| "CPUOffloadWrapper", | |
| "DataProxyConfig", | |
| "EngineConfig", | |
| "EvaluationConfig", | |
| "Modality", | |
| "VarlenHandler", | |
| "get_arch_memory", | |
| "is_hopper_arch", | |
| "parse_config", | |
| ] | |