diff --git a/src/llama_leap.nim b/src/llama_leap.nim
index 982313a..98afcf5 100644
--- a/src/llama_leap.nim
+++ b/src/llama_leap.nim
@@ -23,7 +23,11 @@ type
     repeat_penalty*: Option[float32]
     temperature*: Option[float32]
     seed*: Option[int]
-    # TODO rest
+    stop*: Option[string]
+    tfs_z*: Option[float32]
+    num_predictions*: Option[int]
+    top_k*: Option[int]
+    top_p*: Option[float32]
   GenerateReq* = ref object
     model*: string
     prompt*: string