[Diffusion] [Model] Dreamid omni (#1855)

Signed-off-by: dengyunyang <584797741@qq.com>
This commit is contained in:
dengyunyang authored and GitHub committed 2026-03-13 21:53:06 +08:00
1 parent 92f771d455
commit 3e8cb6c631
17 files changed
+1773 -9

No files matched your search

+1
View File
@@ -322,6 +322,7 @@ class OmniDiffusionConfig:
dtype: torch.dtype = torch.bfloat16
model_config: dict[str, Any] = field(default_factory=dict)
tf_model_config: TransformerConfig = field(default_factory=TransformerConfig)
# Attention