(self, embedding_dim: int, num_embeddings: int)
| 58 | """ |
| 59 | |
| 60 | def __init__(self, embedding_dim: int, num_embeddings: int): |
| 61 | super().__init__() |
| 62 | |
| 63 | self.emb = CombinedTimestepLabelEmbeddings(num_embeddings, embedding_dim) |
| 64 | |
| 65 | self.silu = nn.SiLU() |
| 66 | self.linear = nn.Linear(embedding_dim, 6 * embedding_dim, bias=True) |
| 67 | self.norm = nn.LayerNorm(embedding_dim, elementwise_affine=False, eps=1e-6) |
| 68 | |
| 69 | def forward( |
| 70 | self, |
nothing calls this directly
no test coverage detected