diff --git a/examples/trainer_llm/run.py b/examples/trainer_llm/run.py
index ae22c46..80b1970 100644
--- a/examples/trainer_llm/run.py
+++ b/examples/trainer_llm/run.py
@@ -55,6 +55,9 @@ def parse_args() -> argparse.Namespace:
 
     arg_parser = argparse.ArgumentParser()
     arg_parser.add_argument("--version", action="store_true")
+    arg_parser.add_argument("--output-path", type=pathlib.Path)
+    arg_parser.add_argument("--config", type=pathlib.Path)
+
     args = arg_parser.parse_args()
 
     # If no --version, run parsing of trainign/decomposition arguments
@@ -72,6 +75,7 @@ def parse_args() -> argparse.Namespace:
             required=True,
         )
         args = arg_parser.parse_args()
+        args.version = False
 
     return args
 
diff --git a/examples/trainer_llm/version.py b/examples/trainer_llm/version.py
index ac904f4..dc0b11d 100644
--- a/examples/trainer_llm/version.py
+++ b/examples/trainer_llm/version.py
@@ -1,5 +1,5 @@
 # This is parsed by setup.py, so we need to stick to str -> int parsing
-__version__ = "0.0.64"
+__version__ = "0.0.65"
 
 _ver_major = int(__version__.split(".")[0])
 _ver_minor = int(__version__.split(".")[1])