Method
__init__
(self, embedding_dim, max_positions=10000, scale=1.0)
Source from the content-addressed store, hash-verified
| 42 | """ Code adapted from https://github.com/hojonathanho/diffusion/blob/master/diffusion_tf/nn.py """ |
| 43 | |
| 44 | def __init__(self, embedding_dim, max_positions=10000, scale=1.0): |
| 45 | super().__init__() |
| 46 | self.embedding_dim = embedding_dim |
| 47 | self.max_positions = max_positions |
| 48 | self.scale = scale |
| 49 | |
| 50 | def forward(self, timesteps): |
| 51 | timesteps *= self.scale |
Callers
nothing calls this directly
Tested by
no test coverage detected