1.7.0

residual vq should only do one projection to the codebook dim, and on…

1.6.32

allow for freezing codebook on forward, also convenience setter for c…

1.6.31

init self.embed_avg correctly during kmeans init

1.6.30

manually calculate cdist, as observed NaNs using torch.cdist in anoth…

1.6.29

manually calculate cdist, as observed NaNs using torch.cdist in anoth…

1.6.28

address yet more edge cases with masked tokens, losses, and kmeans init

1.6.27

address yet more edge cases with masked tokens, losses, and kmeans init

1.6.26

when doing ema update in the presence of masked tokens, make sure to …

1.6.25

handle some more edge cases with variable lengthed sequences

1.6.24

1.6.24