Parse Encodec hyperparameters.
(config, outfile, use_f16)
| 57 | |
| 58 | |
| 59 | def parse_codec_hparams(config, outfile, use_f16): |
| 60 | """Parse Encodec hyperparameters.""" |
| 61 | in_channels = config["audio_channels"] |
| 62 | hidden_dim = config["hidden_size"] |
| 63 | n_filters = config["num_filters"] |
| 64 | kernel_size = config["kernel_size"] |
| 65 | residual_kernel_size = config["residual_kernel_size"] |
| 66 | n_bins = config["codebook_size"] |
| 67 | bandwidth = 24 # TODO: hardcoded |
| 68 | sr = config["sampling_rate"] |
| 69 | ftype = int(use_f16) |
| 70 | |
| 71 | outfile.write(struct.pack("i", in_channels)) |
| 72 | outfile.write(struct.pack("i", hidden_dim)) |
| 73 | outfile.write(struct.pack("i", n_filters)) |
| 74 | outfile.write(struct.pack("i", kernel_size)) |
| 75 | outfile.write(struct.pack("i", residual_kernel_size)) |
| 76 | outfile.write(struct.pack("i", n_bins)) |
| 77 | outfile.write(struct.pack("i", bandwidth)) |
| 78 | outfile.write(struct.pack("i", sr)) |
| 79 | outfile.write(struct.pack("i", ftype)) |
| 80 | |
| 81 | |
| 82 | def parse_hparams(config, prefix, outfile, use_f16): |