Skip to content

Commit 7dbcc06

Browse files
committed
set n_predict for summary test
1 parent 2349827 commit 7dbcc06

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/cpp/llamalib_tests.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1034,6 +1034,7 @@ void test_overflow(LLMService *llm_service, int n_ctx)
10341034
{
10351035
std::string system_prompt = "Say hi";
10361036
LLMAgent *agent = new LLMAgent(llm_service, system_prompt);
1037+
agent->set_completion_params({{"seed", 0}, {"n_predict", 30}, {"temperature", 0}});
10371038

10381039
int num_left = n_ctx - llm_service->tokenize(system_prompt).size();
10391040
int num_messages = 8;

0 commit comments

Comments
 (0)