interface

monofuel · Jan 13, 2024 · 41c1395 · 41c1395
1 parent 4ad3efc
commit 41c1395
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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