| 439 | } |
| 440 | |
| 441 | StableAudioRfFeedForwardWeights load_feed_forward( |
| 442 | core::BackendWeightStore & store, |
| 443 | const assets::TensorSource & source, |
| 444 | const std::string & prefix, |
| 445 | const StableAudioConfig & config, |
| 446 | assets::TensorStorageType storage_type) { |
| 447 | StableAudioRfFeedForwardWeights weights; |
| 448 | weights.in_proj = load_linear( |
| 449 | store, |
| 450 | source, |
| 451 | prefix + ".ff.0.proj", |
| 452 | storage_type, |
| 453 | ff_inner_dim(config) * 2, |
| 454 | config.embed_dim, |
| 455 | true); |
| 456 | weights.out_proj = load_linear( |
| 457 | store, |
| 458 | source, |
| 459 | prefix + ".ff.2", |
| 460 | storage_type, |
| 461 | config.embed_dim, |
| 462 | ff_inner_dim(config), |
| 463 | true); |
| 464 | return weights; |
| 465 | } |
| 466 | |
| 467 | StableAudioRfLayerWeights load_layer( |
| 468 | core::BackendWeightStore & store, |
no test coverage detected