tzhouam a97409efef Add dtype mapping method in OmniProcessor ...

- Introduced a static method `_dtype_to_name` to map PyTorch data types to their string representations.
- Updated the `dtype_str` assignments in the `OmniProcessor` class to utilize the new mapping method for improved readability and maintainability.

2025-10-24 11:40:43 +08:00

..

__init__.py

Add Omni model components and input processing for hidden states support

2025-10-22 15:16:25 +08:00

arg_utils.py

Add Omni model components and input processing for hidden states support

2025-10-22 15:16:25 +08:00

diffusion_engine.py

add back the diffusion engine as a placeholder for the actural diffusion engine in the future

2025-10-23 10:04:27 +08:00

output_processor.py

update ar_dit_test

2025-09-28 21:58:48 +08:00

processor.py

Add dtype mapping method in OmniProcessor

2025-10-24 11:40:43 +08:00