| 152 | } |
| 153 | validate_config(parsed, family); |
| 154 | RoformerArchitectureConfig config; |
| 155 | config.family = std::string(family); |
| 156 | config.sample_rate = json::require_i32(parsed, "sample_rate"); |
| 157 | config.stereo = json::optional_bool(parsed, "stereo", true); |
| 158 | config.channels = config.stereo ? 2 : 1; |
| 159 | config.chunk_size = json::require_i32(parsed, "chunk_size"); |
| 160 | config.inference_batch_size = json::optional_i32(parsed, "batch_size", 1); |
| 161 | config.inference_num_overlap = json::require_i32(parsed, "num_overlap"); |
| 162 | config.inference_normalize = json::optional_bool(parsed, "normalize", false); |
| 163 | config.dim = json::require_i32(parsed, "dim"); |
| 164 | config.depth = json::require_i32(parsed, "depth"); |
| 165 | config.num_stems = json::require_i32(parsed, "num_stems"); |
no test coverage detected