Staticaliza
commited on
Commit
•
e2ad91e
1
Parent(s):
514b2f9
Update app.py
Browse files
app.py
CHANGED
@@ -33,9 +33,10 @@ for model_name, model_endpoint in API_ENDPOINTS.items():
|
|
33 |
|
34 |
def format(instruction, history, input, wrap):
|
35 |
sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
|
|
|
36 |
formatted_history = "".join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}\n" for message in history)
|
37 |
formatted_input = f"{sy_l}INSTRUCTIONS: {instruction}{sy_r}\n{formatted_history}{sy_l}{input}{sy_r}\n{sy_l}"
|
38 |
-
return f"{formatted_input}{
|
39 |
|
40 |
def predict(access_key, instruction, history, input, wrap, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):
|
41 |
|
|
|
33 |
|
34 |
def format(instruction, history, input, wrap):
|
35 |
sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
|
36 |
+
wrapped_input = wrap % ("")
|
37 |
formatted_history = "".join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}\n" for message in history)
|
38 |
formatted_input = f"{sy_l}INSTRUCTIONS: {instruction}{sy_r}\n{formatted_history}{sy_l}{input}{sy_r}\n{sy_l}"
|
39 |
+
return f"{formatted_input}{wrapped_input}", formatted_input
|
40 |
|
41 |
def predict(access_key, instruction, history, input, wrap, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):
|
42 |
|