anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat(minimax-h3): enable 2x RTX 4090/5090 with DLO (#5764)

Browse Source 
Signed-off-by: lishunyang12 <lishunyang03@gmail.com>
Signed-off-by: lishunyang12 <lishunyang12@163.com>
Co-authored-by: lishunyang12 <lishunyang03@gmail.com>
This commit is contained in:
SYLARandlishunyang12 authored and GitHub committed 2026-08-06 09:02:29 +08:00
1 parent b4581b293b
commit 1c2a81f6d8
28 files changed
+1658 -162
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/offloader/module_collector.py
+3 -1
View File
@@ -19,6 +19,7 @@ class PipelineModules:
encoders: list[nn.Module]
encoder_names: list[str]
vaes: list[nn.Module]
vae_names: list[str] = field(default_factory=list)
resident_modules: list[nn.Module] = field(default_factory=list)
resident_names: list[str] = field(default_factory=list)
@@ -153,7 +154,7 @@ class ModuleDiscovery:
dit_modules, dit_names = ModuleDiscovery._collect_modules(pipeline, dit_attrs, warn_missing=declared)
encoders, encoder_names = ModuleDiscovery._collect_modules(pipeline, enc_attrs, warn_missing=declared)
vaes, _ = ModuleDiscovery._collect_modules(pipeline, vae_attrs, warn_missing=declared)
vaes, vae_names = ModuleDiscovery._collect_modules(pipeline, vae_attrs, warn_missing=declared)
residents, resident_names = ModuleDiscovery._collect_modules(pipeline, res_attrs, warn_missing=declared)
return PipelineModules(
@@ -163,5 +164,6 @@ class ModuleDiscovery:
encoder_names=encoder_names,
vaes=vaes,
resident_modules=residents,
vae_names=vae_names,
resident_names=resident_names,
)