3 Commits

Include renames

Author	SHA1	Message	Date
tzhouam	2cc33bb3ea	Fix tensor handling in DiffusionModelRunner for multimodal outputs ... - Updated assertions and tensor management to correctly handle multimodal outputs in the `DiffusionModelRunner` class. - Ensured that the output tensors are processed based on the type of multimodal outputs received, improving robustness and clarity in the code.	2025-10-22 17:38:49 +08:00
tzhouam	811057ce11	Refactor GPU diffusion model runner and worker ... - Introduced `DiffusionModelRunner` class to handle diffusion processes, extending `OmniGPUModelRunner`. - Updated `DiffusionGPUWorker` to initialize device settings and manage memory more effectively. - Enhanced model execution with support for multimodal inputs and improved tensor management. - Removed legacy code and streamlined the generation process for better performance.	2025-10-22 15:30:46 +08:00
hsliu	81f04d3cf9	update ar_dit_test	2025-09-28 21:58:48 +08:00