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

oops 2

parent 02ba284b

audiolm_pytorch_demo_laion.py

+3 −3
Original line number Diff line number Diff line
@@ -98,9 +98,9 @@ encodec = EncodecWrapper() 


# 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}",

    checkpoint_path = None,

    # use_mert = True,

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

    # checkpoint_path = None,

    kmeans_path = f"{prefix}/{hubert_quantizer}"

)