| 36 | config.use_wave_decoder = yaml::require_bool(parsed, "model.init_args.config.use_wave_decoder"); |
| 37 | config.wave_upsample_factor = yaml::require_int(parsed, "model.init_args.config.wave_upsample_factor"); |
| 38 | config.wave_decoder_dim = yaml::require_int(parsed, "model.init_args.config.wave_decoder_dim"); |
| 39 | config.wave_resnet_num_blocks = yaml::require_int(parsed, "model.init_args.config.wave_resnet_num_blocks"); |
| 40 | config.wave_resnet_num_groups = yaml::require_int(parsed, "model.init_args.config.wave_resnet_num_groups"); |
| 41 | config.wave_upsampler_factors = yaml::require_list_int(parsed, "model.init_args.config.wave_upsampler_factors"); |
| 42 | config.wave_upsampler_kernel_sizes = yaml::require_list_int(parsed, "model.init_args.config.wave_upsampler_kernel_sizes"); |
| 43 | |
| 44 | config.local_encoder_dim = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.dim"); |
| 45 | config.local_encoder_layers = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.n_layers"); |
| 46 | config.local_encoder_heads = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.n_heads"); |
| 47 | config.local_encoder_window_size = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.window_size"); |
| 48 | config.local_encoder_max_seq_len = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.max_seq_len"); |
| 49 | config.quantizer_input_dim = yaml::require_int(parsed, "model.init_args.local_quantizer.init_args.input_dim"); |
no test coverage detected