(self, embedding_dim: int, use_additional_conditions: bool = False)
| 244 | """ |
| 245 | |
| 246 | def __init__(self, embedding_dim: int, use_additional_conditions: bool = False): |
| 247 | super().__init__() |
| 248 | |
| 249 | self.emb = PixArtAlphaCombinedTimestepSizeEmbeddings( |
| 250 | embedding_dim, size_emb_dim=embedding_dim // 3, use_additional_conditions=use_additional_conditions |
| 251 | ) |
| 252 | |
| 253 | self.silu = nn.SiLU() |
| 254 | self.linear = nn.Linear(embedding_dim, 6 * embedding_dim, bias=True) |
| 255 | |
| 256 | def forward( |
| 257 | self, |
nothing calls this directly
no test coverage detected