| 255 | assets::require_tensor_shape( |
| 256 | weights, |
| 257 | "model.prediction_head.final_layer.linear.weight", |
| 258 | {config.diffusion_head.latent_size, decoder.hidden_size}); |
| 259 | |
| 260 | assets::require_tensor_shape( |
| 261 | weights, |
| 262 | "model.acoustic_tokenizer.encoder.downsample_layers.0.0.conv.conv.weight", |
| 263 | {config.acoustic_tokenizer.encoder_n_filters, config.acoustic_tokenizer.channels, 7}); |
| 264 | assets::require_tensor_shape( |
| 265 | weights, |
| 266 | "model.acoustic_tokenizer.decoder.head.conv.conv.weight", |
| 267 | {config.acoustic_tokenizer.channels, config.acoustic_tokenizer.decoder_n_filters, 7}); |
no test coverage detected