Commit a5dbd73d authored by Leon Wu's avatar Leon Wu
Browse files

note to self

parent 29883e48

audiolm_pytorch_demo_laion.py

+1 −0
Original line number Diff line number Diff line
@@ -97,6 +97,7 @@ encodec = EncodecWrapper() 
#############



raise AssertionError("don't forget to use_mert when the encodec_support PR is merged")

raise AssertionError("note to self, try larger batch size and grad update https://github.com/lucidrains/audiolm-pytorch/discussions/107#discussioncomment-5373414")

wav2vec = HubertWithKmeans(

    # use_mert = True,

    checkpoint_path = f"{prefix}/{hubert_ckpt}",