Can you release the codes for getting discrete units?

[isruihu]

Hi there, Good job! Do you have a plan to release the codes for getting discrete units (using Hubert and a K-means model)? or give a guidance on where to download the required encoder model.

thanks!