add wavebeat ckpt to finetuned confs, rm temperature typing

scripts/exp/fine_tune.py

@@ -53,6 +53,7 @@ def fine_tune(audio_files_or_folders: List[str], name: str):
 
         "Interface.coarse2fine_ckpt": f"./models/vampnet/c2f.pth",
         "Interface.coarse2fine_lora_ckpt": f"./runs/{name}/c2f/latest/lora.pth",
+        "Interface.wavebeat_ckpt": "./models/wavebeat.pth",
 
         "Interface.codec_ckpt": "./models/vampnet/codec.pth",
         "AudioLoader.sources": [audio_files_or_folders],

vampnet/modules/transformer.py

@@ -581,7 +581,7 @@ class VampNet(at.ml.BaseModel):
         sampling_steps: int = 24,
         start_tokens: Optional[torch.Tensor] = None,
         mask: Optional[torch.Tensor] = None,
-        temperature: Union[float, Tuple[float, float]] = 2.5,
+        temperature: float = 2.5,
         typical_filtering=False,
         typical_mass=0.2,
         typical_min_tokens=1,
@@ -592,7 +592,7 @@ class VampNet(at.ml.BaseModel):
         #####################
         # resolve temperature #
         #####################
-        assert isinstance(temperature, float)
+
         logging.debug(f"temperature: {temperature}")