diff --git a/mofa/generator.py b/mofa/generator.py
index 030e3296..75add995 100644
--- a/mofa/generator.py
+++ b/mofa/generator.py
@@ -10,7 +10,6 @@
 import os
 
 def train_generator(
-        config: str="config.yaml",
         # starting_model: str | Path,
         examples: str|list[MOFRecord]="../argonne_gnn_gitlab/DiffLinker/data/geom/datasets",
         num_epochs: int=10