Yuanheng Zhao 82f63a4403 [BugFix] Enable CPU offloading and Cache-DiT together on Diffusion Model (#1723) ...

Signed-off-by: Yuanheng Zhao <jonathan.zhaoyh@gmail.com>
Signed-off-by: yuanheng <jonathan.zhaoyh@gmail.com>

2026-03-09 22:25:41 +08:00

..

stage_configs

[Bagel]: Support multistage img2img (#1669)

2026-03-09 22:23:50 +08:00

__init__.py

[CI] Qwen3-Omni online test (#257)

2025-12-12 14:25:52 +08:00

test_bagel_text2img.py

[Feature]: Support cfg kv-cache transfer in multi-stage (#1422)

2026-03-02 05:06:28 +00:00

test_cache_dit.py

[CI] Add pytest markers to current tests and update the doc. (#577)

2026-02-09 16:49:45 +08:00

test_diffusion_cpu_offload.py

[BugFix] Enable CPU offloading and Cache-DiT together on Diffusion Model (#1723)

2026-03-09 22:25:41 +08:00

test_diffusion_layerwise_offload.py

[BugFix] Enable CPU offloading and Cache-DiT together on Diffusion Model (#1723)

2026-03-09 22:25:41 +08:00

test_diffusion_lora.py

Replace hard-coded cuda generator with current_omni_platform.device_type (#1677)

2026-03-05 06:53:30 +00:00

test_mammoth_moda2.py

add support for MammothModa2 model (#336)

2026-03-04 22:52:56 +08:00

test_ovis_image.py

[Misc] Migrate L1 tests to use pytest-mock (#1315)

2026-02-26 17:08:39 +08:00

test_qwen2_5_omni.py

[Test] Solving the Issue of Whisper Model's GPU Memory Not Being Successfully Cleared and the Occasional Accuracy Problem of the Qwen3-omni Model Test (#1744)

2026-03-09 21:52:56 +08:00

test_qwen3_omni.py

[Test] Solving the Issue of Whisper Model's GPU Memory Not Being Successfully Cleared and the Occasional Accuracy Problem of the Qwen3-omni Model Test (#1744)

2026-03-09 21:52:56 +08:00

test_sequence_parallel.py

[CI] Reduce the time for Diffusion Sequence Parallelism Test (#1283)

2026-02-10 01:48:04 +13:00

test_stable_audio_model.py

Replace hard-coded cuda generator with current_omni_platform.device_type (#1677)

2026-03-05 06:53:30 +00:00

test_t2i_model.py

[Bugfix] remove Tongyi-MAI/Z-Image-Turbo related test from L2 ci (#1348)

2026-02-12 18:44:05 +08:00

test_t2v_model.py

[XPU] Fix XPU UTs for basic coverage (#1164)

2026-02-05 15:04:57 +00:00

test_teacache.py

Replace hard-coded cuda generator with current_omni_platform.device_type (#1677)

2026-03-05 06:53:30 +00:00

test_vae_decode_parallelism.py

[FEATURE] Tile/Patch parallelism refactor for easily support other models (#1366)

2026-03-03 15:48:51 +08:00

test_zimage_parallelism.py

[Bugfix] remove Tongyi-MAI/Z-Image-Turbo related test from L2 ci (#1348)

2026-02-12 18:44:05 +08:00

utils.py

[Feature]Basic version of supporting streaming output (#367)

2026-01-01 03:02:04 +13:00