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

..

__init__.py

Initial vLLM-omni repository structure

2025-09-11 14:32:51 +08:00

gpu_diffusion_model_runner.py

Fix tensor handling in DiffusionModelRunner for multimodal outputs

2025-10-22 17:38:49 +08:00

gpu_diffusion_worker.py

Refactor GPU diffusion model runner and worker

2025-10-22 15:30:46 +08:00