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

switch over to MERT in demo script

parent fd21fd44

audiolm_pytorch_demo_laion.py

+2 −1
Original line number Diff line number Diff line
@@ -91,7 +91,8 @@ make_placeholder_dataset() 
#############



wav2vec = HubertWithKmeans(

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

    use_mert = True,

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

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

)