LAC!

.gitignore

@@ -179,4 +179,5 @@ gradio-outputs/
 models/
 samples*/
 models-all/
-models.zip
+models.zip
+.git-old

scripts/exp/train.py

@@ -20,7 +20,7 @@ import vampnet
 from vampnet.modules.transformer import VampNet
 from vampnet.util import codebook_unflatten, codebook_flatten
 from vampnet import mask as pmask
-from lac.model.lac import LAC
+from dac.model.dac import DAC
 
 
 # Enable cudnn autotuner to speed up training
@@ -109,7 +109,7 @@ def load(
     load_weights: bool = False,
     fine_tune_checkpoint: Optional[str] = None,
 ):
-    codec = LAC.load(args["codec_ckpt"], map_location="cpu")
+    codec = DAC.load(args["codec_ckpt"], map_location="cpu")
     codec.eval()
 
     model, v_extra = None, {}

vampnet/interface.py

@@ -11,7 +11,7 @@ from .modules.transformer import VampNet
 from .beats import WaveBeat
 from .mask import *
 
-from lac.model.lac import LAC
+from dac.model.dac import DAC
 
 
 def signal_concat(
@@ -63,7 +63,7 @@ class Interface(torch.nn.Module):
     ):
         super().__init__()
         assert codec_ckpt is not None, "must provide a codec checkpoint"
-        self.codec = LAC.load(Path(codec_ckpt))
+        self.codec = DAC.load(Path(codec_ckpt))
         self.codec.eval()
         self.codec.to(device)