[diffusion] z-image support (#149)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
Co-authored-by: Samit <285365963@qq.com>
This commit is contained in:
Jiangyun ZhuandSamit authored and GitHub committed 2025-12-03 02:16:36 +00:00
1 parent 5a0d7882c5
commit 7a6d2fe3f6
9 files changed
+1344 -25

No files matched your search

+2
View File
@@ -21,6 +21,8 @@ class OmniDiffusionConfig:
model_class_name: str | None = None
dtype: torch.dtype = torch.bfloat16
# Attention
# attention_backend: str = None