| 41 | |
| 42 | @dataclass |
| 43 | class DiffusionModelConfig: |
| 44 | cross_attention_dim: int = 2048 |
| 45 | dropout: float = 0.0 |
| 46 | final_dropout: bool = True |
| 47 | interleave_self_attention: bool = True |
| 48 | norm_type: str = "ada_norm" |
| 49 | num_layers: int = 16 |
| 50 | output_dim: int = 1024 |
| 51 | positional_embeddings: Optional[str] = None |
| 52 | |
| 53 | |
| 54 | @dataclass |