| 306 | } |
| 307 | |
| 308 | ResidualUnitWeights load_residual_unit( |
| 309 | core::BackendWeightStore & store, |
| 310 | const assets::TensorSource & source, |
| 311 | const std::string & prefix, |
| 312 | int64_t channels, |
| 313 | assets::TensorStorageType storage_type) { |
| 314 | ResidualUnitWeights out = {}; |
| 315 | out.snake1 = load_snake_exact(store, source, prefix + ".snake1", channels); |
| 316 | out.conv1 = load_weight_norm_conv1d(store, source, prefix + ".conv1", channels, channels, 7, 1, 3, 1, storage_type, true); |
| 317 | out.snake2 = load_snake_exact(store, source, prefix + ".snake2", channels); |
| 318 | out.conv2 = load_weight_norm_conv1d(store, source, prefix + ".conv2", channels, channels, 1, 1, 0, 1, storage_type, true); |
| 319 | return out; |
| 320 | } |
| 321 | |
| 322 | VAEDecoderWeights load_vae_decoder_weights( |
| 323 | const AceStepAssets & assets, |
no test coverage detected