Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ def get_usernames(model: str):
|
|
| 14 |
Returns:
|
| 15 |
(str, str, str, str): pre-prompt, username, bot name, separator
|
| 16 |
"""
|
| 17 |
-
if model in ("OpenAssistant/oasst-sft-1-pythia-12b"
|
| 18 |
return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
|
| 19 |
|
| 20 |
def predict(
|
|
@@ -52,7 +52,7 @@ def predict(
|
|
| 52 |
|
| 53 |
partial_words = ""
|
| 54 |
|
| 55 |
-
if model in ("OpenAssistant/oasst-sft-1-pythia-12b"
|
| 56 |
iterator = client.generate_stream(
|
| 57 |
total_inputs,
|
| 58 |
typical_p=typical_p,
|
|
@@ -109,7 +109,7 @@ def radio_on_change(
|
|
| 109 |
repetition_penalty,
|
| 110 |
watermark,
|
| 111 |
):
|
| 112 |
-
if value in ("OpenAssistant/oasst-sft-1-pythia-12b"
|
| 113 |
typical_p = typical_p.update(value=0.2, visible=True)
|
| 114 |
top_p = top_p.update(visible=False)
|
| 115 |
top_k = top_k.update(visible=False)
|
|
@@ -139,12 +139,6 @@ def radio_on_change(
|
|
| 139 |
title = """<h1 align="center">LLM Chat</h1>"""
|
| 140 |
description = """LLM Chat predložak:
|
| 141 |
```
|
| 142 |
-
User: <utterance>
|
| 143 |
-
Assistant: <utterance>
|
| 144 |
-
User: <utterance>
|
| 145 |
-
Assistant: <utterance>
|
| 146 |
-
...
|
| 147 |
-
```
|
| 148 |
Mjenjajući predložak.
|
| 149 |
"""
|
| 150 |
|
|
|
|
| 14 |
Returns:
|
| 15 |
(str, str, str, str): pre-prompt, username, bot name, separator
|
| 16 |
"""
|
| 17 |
+
if model in ("OpenAssistant/oasst-sft-1-pythia-12b"):
|
| 18 |
return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
|
| 19 |
|
| 20 |
def predict(
|
|
|
|
| 52 |
|
| 53 |
partial_words = ""
|
| 54 |
|
| 55 |
+
if model in ("OpenAssistant/oasst-sft-1-pythia-12b"):
|
| 56 |
iterator = client.generate_stream(
|
| 57 |
total_inputs,
|
| 58 |
typical_p=typical_p,
|
|
|
|
| 109 |
repetition_penalty,
|
| 110 |
watermark,
|
| 111 |
):
|
| 112 |
+
if value in ("OpenAssistant/oasst-sft-1-pythia-12b"):
|
| 113 |
typical_p = typical_p.update(value=0.2, visible=True)
|
| 114 |
top_p = top_p.update(visible=False)
|
| 115 |
top_k = top_k.update(visible=False)
|
|
|
|
| 139 |
title = """<h1 align="center">LLM Chat</h1>"""
|
| 140 |
description = """LLM Chat predložak:
|
| 141 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 142 |
Mjenjajući predložak.
|
| 143 |
"""
|
| 144 |
|