Update app.py
Browse files
app.py
CHANGED
@@ -88,7 +88,7 @@ def load_models(inp):
|
|
88 |
if "mixtral" in models[inp].lower():
|
89 |
prompt_out="[INST] USER_INPUT [/INST]"
|
90 |
return out_box,prompt_out, model_state
|
91 |
-
|
92 |
prompt_out="GPT4 Correct User: USER_INPUT<|end_of_turn|>GPT4 Correct Assistant: "
|
93 |
return out_box,prompt_out, model_state
|
94 |
else:
|
|
|
88 |
if "mixtral" in models[inp].lower():
|
89 |
prompt_out="[INST] USER_INPUT [/INST]"
|
90 |
return out_box,prompt_out, model_state
|
91 |
+
if "openchat" in models[inp].lower():
|
92 |
prompt_out="GPT4 Correct User: USER_INPUT<|end_of_turn|>GPT4 Correct Assistant: "
|
93 |
return out_box,prompt_out, model_state
|
94 |
else:
|