update

levo_inference.py

@@ -28,6 +28,7 @@ class LeVoInference(torch.nn.Module):
         OmegaConf.register_new_resolver("load_yaml", lambda x: list(OmegaConf.load(x)))
 
         self.cfg = OmegaConf.load(cfg_path)
+        self.cfg.mode = 'inference'
         self.max_duration = self.cfg.max_dur
 
         # Define model or load pretrained model