Spaces:
Sleeping
Sleeping
DylanZimmer commited on
Commit ·
6f5eb81
1
Parent(s): dc94fda
Lower for testing
Browse files
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=
|
| 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=
|
| 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,
|