Export without loading dataset

src/python/larynx_train/export_generator.py

@@ -35,7 +35,7 @@ def main():
     args.output = Path(args.output)
     args.output.parent.mkdir(parents=True, exist_ok=True)
 
-    model = VitsModel.load_from_checkpoint(args.checkpoint)
+    model = VitsModel.load_from_checkpoint(args.checkpoint, dataset=None)
     model_g = model.model_g
 
     # Inference only

src/python/larynx_train/export_onnx.py

@@ -39,7 +39,7 @@ def main():
     args.output = Path(args.output)
     args.output.parent.mkdir(parents=True, exist_ok=True)
 
-    model = VitsModel.load_from_checkpoint(args.checkpoint)
+    model = VitsModel.load_from_checkpoint(args.checkpoint, dataset=None)
     model_g = model.model_g
 
     num_symbols = model_g.n_vocab

src/python/larynx_train/export_torchscript.py

@@ -37,7 +37,7 @@ def main():
     args.output = Path(args.output)
     args.output.parent.mkdir(parents=True, exist_ok=True)
 
-    model = VitsModel.load_from_checkpoint(args.checkpoint)
+    model = VitsModel.load_from_checkpoint(args.checkpoint, dataset=None)
     model_g = model.model_g
 
     num_symbols = model_g.n_vocab

src/python/larynx_train/vits/lightning.py

@@ -124,6 +124,10 @@ class VitsModel(pl.LightningModule):
         num_test_examples: int,
         max_phoneme_ids: Optional[int] = None,
     ):
+        if self.hparams.dataset is None:
+            _LOGGER.debug("No dataset to load")
+            return
+
         full_dataset = LarynxDataset(
             self.hparams.dataset, max_phoneme_ids=max_phoneme_ids
         )