DylanZimmer commited on
Commit
6f5eb81
·
1 Parent(s): dc94fda

Lower for testing

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ def apply_tiny_chat_template(messages):
33
  prompt += f"<|response|> {content}\n"
34
  return prompt
35
 
36
- def chat_fxn_caller(message, history, system_prompt=system_prompt_default, temperature=0.6, top_p=0.95, max_tokens=256):
37
  messages = []
38
  if system_prompt.strip():
39
  messages.append({"role": "system", "content": system_prompt})
@@ -50,7 +50,7 @@ def chat_fxn_caller(message, history, system_prompt=system_prompt_default, tempe
50
  prompt,
51
  do_sample=True,
52
  max_new_tokens=max_tokens,
53
- top_k=50,
54
  temperature=temperature,
55
  num_return_sequences=1,
56
  repetition_penalty=1.1,
 
33
  prompt += f"<|response|> {content}\n"
34
  return prompt
35
 
36
+ def chat_fxn_caller(message, history, system_prompt=system_prompt_default, temperature=0.6, top_p=0.95, max_tokens=128):
37
  messages = []
38
  if system_prompt.strip():
39
  messages.append({"role": "system", "content": system_prompt})
 
50
  prompt,
51
  do_sample=True,
52
  max_new_tokens=max_tokens,
53
+ top_k=30,
54
  temperature=temperature,
55
  num_return_sequences=1,
56
  repetition_penalty=1.1,