[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:
1 parent
5a0d7882c5
commit
7a6d2fe3f6
9 files changed
+1344
-25
No files matched your search
@@ -21,6 +21,8 @@ class OmniDiffusionConfig:
|
||||
|
||||
model_class_name: str | None = None
|
||||
|
||||
dtype: torch.dtype = torch.bfloat16
|
||||
|
||||
# Attention
|
||||
# attention_backend: str = None
|
||||
|
||||
|
||||
Reference in new issue
Block a user