| 221 | } |
| 222 | |
| 223 | TimestepEmbeddingModule::TimestepEmbeddingModule(TimestepEmbeddingConfig config) : config_(config) { |
| 224 | require_positive(config_.frequency_embedding_size, "TimestepEmbeddingConfig.frequency_embedding_size"); |
| 225 | require_positive(config_.hidden_size, "TimestepEmbeddingConfig.hidden_size"); |
| 226 | if (config_.frequency_embedding_size % 2 != 0) { |
| 227 | throw std::runtime_error("TimestepEmbeddingConfig.frequency_embedding_size must be even"); |
| 228 | } |
| 229 | } |
| 230 | |
| 231 | core::TensorValue TimestepEmbeddingModule::build( |
| 232 | core::ModuleBuildContext & ctx, |
nothing calls this directly
no test coverage detected