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 |
|