[Diffusion] [Model] Dreamid omni (#1855)
Signed-off-by: dengyunyang <584797741@qq.com>
This commit is contained in:
1 parent
92f771d455
commit
3e8cb6c631
17 files changed
+1773
-9
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user