| 188 | store, |
| 189 | source, |
| 190 | prefix + ".emb.timestep_embedder.linear_1", |
| 191 | storage_type, |
| 192 | hidden_size, |
| 193 | 512, |
| 194 | true); |
| 195 | weights.timestep_linear_2 = binding::linear_from_source( |
| 196 | store, |
| 197 | source, |
| 198 | prefix + ".emb.timestep_embedder.linear_2", |
| 199 | storage_type, |
| 200 | hidden_size, |
| 201 | hidden_size, |
| 202 | true); |
| 203 | weights.linear = binding::linear_from_source( |
| 204 | store, |
| 205 | source, |
| 206 | prefix + ".linear", |
| 207 | storage_type, |
| 208 | 6 * hidden_size, |
| 209 | hidden_size, |
| 210 | true); |
| 211 | return weights; |
| 212 | } |
| 213 | |
| 214 | HeartCodecTransformerBlockWeights load_transformer_block( |
| 215 | core::BackendWeightStore & store, |
| 216 | const assets::TensorSource & source, |
| 217 | const std::string & prefix, |
| 218 | assets::TensorStorageType storage_type, |
no test coverage detected