allow for specifying which layer of hubert to use

def exists(val):

    return val is not None

def default(val, d):

    return val if exists(val) else d

class HubertWithKmeans(nn.Module):

    """

    checkpoint and kmeans can be downloaded at https://github.com/facebookresearch/fairseq/tree/main/examples/hubert

        checkpoint_path,

        kmeans_path,

        target_sample_hz = 16000,

        seq_len_multiple_of = None

        seq_len_multiple_of = None,

        output_layer = 9

    ):

        super().__init__()

        self.target_sample_hz = target_sample_hz

        if exists(self.seq_len_multiple_of):

            wav_input = curtail_to_multiple(wav_input, self.seq_len_multiple_of)

        embed = self.model(wav_input, features_only = True)

        embed = self.model(wav_input, features_only = True, output_layer = self.output_layer)

        embed, packed_shape = pack([embed['x']], '* d')

        codebook_indices = self.kmeans.predict(embed.cpu().detach().numpy())

__version__ = '0.25.0'

__version__ = '0.25.1'