Gao Han 69fd466fba Merge branch 'main' into feat/diffusion-prefetch-protection

2026-06-02 23:22:08 +08:00

..

attention

Add Cosmos3 model support (#3454)

2026-06-02 04:04:31 +08:00

cache

Add Cosmos3 model support (#3454)

2026-06-02 04:04:31 +08:00

distributed

[Diffusion] Support USP and VAE patch parallel for HunyuanVideo 1.5 (#3979)

2026-05-30 14:40:42 +08:00

executor

[Bugfix] Fix DiffusionWorker crash on SIGINT: ensure NCCL/ZMQ cleanup on shutdown (#3872)

2026-05-31 00:13:23 +08:00

hooks

[Feat] Support MagCache (#1287)

2026-05-29 22:26:23 +08:00

layers

[Blackwell] Add CUDNN_ATTN and FLASHINFER_ATTN backends for diffusion (auto-route) (#3079)

2026-05-28 11:09:25 +08:00

lora

[Feature] Enable LoRA adapter injection for BAGEL (#2490)

2026-04-06 19:30:52 +08:00

model_loader

fix: correct logging message format in hub_prefetch.py for better clarity during prefetch attempts

2026-06-02 09:01:47 +00:00

models

Merge branch 'main' into feat/diffusion-prefetch-protection

2026-06-02 23:22:08 +08:00

offloader

[New Model]: Add sensenova u1 support (#3319)

2026-05-11 22:19:00 +08:00

postprocess

[Bugfix] Fix RIFE device selection for CPU-transported videos (#2876)

2026-04-17 19:19:34 +08:00

profiler

[Feature] support sp for hunyuan (#2163)

2026-03-27 17:28:13 +08:00

sched

[Diffusion] Unify diffusion request identity on request_id (#3744)

2026-05-28 14:56:56 +08:00

utils

[Bugfix] enforce max_sequence_length for Qwen-Image and Wan2.2 series before encoding (#2847)

2026-04-17 10:40:36 +08:00

worker

[ROCm] [CI] Bugfix Existing CI cases (#3946)

2026-06-01 22:02:15 +08:00

__init__.py

[Diffusion] Qwen image support (#82)

2025-11-28 14:22:11 +08:00

compile.py

[core] add torch compile for diffusion (#684)

2026-01-09 04:05:44 +13:00

config.py

[Diffusion][Attention] Support per-role attention backend via CLI (#2681)

2026-05-10 11:59:05 +08:00

data.py

[New Model] Add Lance (ByteDance) (#3710)

2026-06-01 16:01:25 +08:00

diffusion_engine.py

Add Cosmos3 model support (#3454)

2026-06-02 04:04:31 +08:00

envs.py

[Refactor] Extract has_flash_attn_pkg in OmniPlatform (#3068)

2026-04-27 04:07:34 +00:00

forward_context.py

[Model] Add Ming-flash-omni-2.0 Image Generation (Diffusion) Stage (#2875)

2026-05-27 23:21:20 +08:00

inline_stage_diffusion_client.py

[Diffusion] Unify diffusion request identity on request_id (#3744)

2026-05-28 14:56:56 +08:00

ipc.py

Add Cosmos3 model support (#3454)

2026-06-02 04:04:31 +08:00

model_metadata.py

[Feature] HunyuanImage-3.0 IT2I: multi-image input + prompt API cleanup (#3444)

2026-05-14 16:58:40 +08:00

registry.py

Add Cosmos3 model support (#3454)

2026-06-02 04:04:31 +08:00

request.py

[Diffusion] Unify diffusion request identity on request_id (#3744)

2026-05-28 14:56:56 +08:00

stage_diffusion_client.py

[large-scale-serving] Integrate OmniCoordinator into stage engine pipeline (#3569)

2026-05-19 12:08:20 +08:00

stage_diffusion_proc.py

[Diffusion] Unify diffusion request identity on request_id (#3744)

2026-05-28 14:56:56 +08:00