Unverified Commit 8344d2c2 authored by Taylor's avatar Taylor Committed by GitHub
Browse files

Update hubert_kmeans.py 

https://github.com/lucidrains/audiolm-pytorch/issues/139
parent 0240f6a6

audiolm_pytorch/hubert_kmeans.py

+1 −0
Original line number Diff line number Diff line
@@ -38,6 +38,7 @@ class HubertWithKmeans(nn.Module): 
        super().__init__()

        self.target_sample_hz = target_sample_hz

        self.seq_len_multiple_of = seq_len_multiple_of

        self.output_layer = output_layer



        model_path = Path(checkpoint_path)

        kmeans_path = Path(kmeans_path)