Add frequency penalty sampling in test

dottxt-ai · Feb 10, 2024 · b31d731 · b31d731
1 parent 422a9e9
commit b31d731
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/generate/test_integration_llamacpp.py b/tests/generate/test_integration_llamacpp.py
@@ -287,7 +287,9 @@ class User(BaseModel):
         value: List[float]
 
     result = generate.json(model, User)(
-        prompt, max_tokens=500, kw_args={"temperature": 0.0, "seed": 0}
+        prompt,
+        max_tokens=500,
+        kw_args={"temperature": 0.0, "seed": 0, "frequency_penalty": 0.5},
     )
     assert isinstance(result, BaseModel)
     assert isinstance(result.id, int)