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

Revert "try dumping original shapes again?" 

This reverts commit 31d2c947.
parent 31d2c947

audiolm_pytorch_demo_laion.py

+3 −3
Original line number Diff line number Diff line
@@ -91,9 +91,9 @@ make_placeholder_dataset() 
#############



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

)