(self, codec: RedCodec)
| 196 | |
| 197 | class RedCodecInfer(RedCodec): |
| 198 | def __init__(self, codec: RedCodec): |
| 199 | super().__init__( |
| 200 | codec.ssl, |
| 201 | codec.ssl_adaptor, |
| 202 | codec.acoustic_encoder, |
| 203 | codec.downsample, |
| 204 | codec.rvq, |
| 205 | codec.upsample, |
| 206 | codec.semantic_decoder, |
| 207 | codec.acoustic_decoder, |
| 208 | ) |
| 209 | |
| 210 | @classmethod |
| 211 | def from_pretrained(cls, conf_path: str, ckpt_path: str) -> "RedCodecInfer": |