Spaces:

arnabdas8901
/

Magnifying_Lens

Sleeping

Arnab Das commited on Nov 15, 2024

Commit

667dd63

1 Parent(s): e646637

bug fix

Files changed (3) hide show

manipulate_model/demo-model/audio/config.yaml CHANGED Viewed

@@ -4,7 +4,7 @@ model:
     name: wavlm
     version: base
     pretrained: true
-    pretrained_path: manipulate_model/encoder_checkpoints/wavlm/WavLM-Base+.pt
     output_layer: 3
   encoder_freeze: false
   decoder:

     name: wavlm
     version: base
     pretrained: true
+    pretrained_path: manipulate_model/encoder_checkpoints/wavlm/WavLM_basePlus_config.pt
     output_layer: 3
   encoder_freeze: false
   decoder:

manipulate_model/encoder/encoder.py CHANGED Viewed

@@ -21,7 +21,7 @@ class Encoder(nn.Module):
                 ckpt = torch.load(
                     config.model.encoder.pretrained_path, map_location="cpu"
                 )
-                cfg = WavLMConfig(ckpt["cfg"])
                 self.encoder = WavLM(cfg)

                 ckpt = torch.load(
                     config.model.encoder.pretrained_path, map_location="cpu"
                 )
+                cfg = WavLMConfig(ckpt)
                 self.encoder = WavLM(cfg)

manipulate_model/encoder_checkpoints/wavlm/WavLM_basePlus_config.pt ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:57196dd3b42133c5807bcd18f34c50e4b782003b018cac21e23d63ec088538a6
+size 1944