Spaces:
Runtime error
Runtime error
Fetching metadata from the HF Docker repository...
Commit
·
8a7a091
1
Parent(s):
6728f78
Add latest OA Pythia (#13)
Browse files- Add latest OA Pythia (f4705757473185eabcce1b7d88b9ee23d1195651)
Co-authored-by: Oliver Stanley <OllieStanley@users.noreply.huggingface.co>
app.py
CHANGED
@@ -23,7 +23,7 @@ def get_usernames(model: str):
23
Returns:
24
(str, str, str, str): pre-prompt, username, bot name, separator
25
"""
26
-
if model (削除) == (削除ここまで) "OpenAssistant/oasst-sft-1-pythia-12b":
27
return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
28
if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
29
return openchat_preprompt, "<human>: ", "<bot>: ", "\n"
@@ -65,7 +65,7 @@ def predict(
65
66
partial_words = ""
67
68
-
if model (削除) == (削除ここまで) "OpenAssistant/oasst-sft-1-pythia-12b":
69
iterator = client.generate_stream(
70
total_inputs,
71
typical_p=typical_p,
@@ -122,7 +122,7 @@ def radio_on_change(
122
repetition_penalty,
123
watermark,
124
):
125
-
if value (削除) == (削除ここまで) "OpenAssistant/oasst-sft-1-pythia-12b":
126
typical_p = typical_p.update(value=0.2, visible=True)
127
top_p = top_p.update(visible=False)
128
top_k = top_k.update(visible=False)
@@ -187,8 +187,9 @@ with gr.Blocks(
187
gr.Markdown(text_generation_inference, visible=True)
188
with gr.Column(elem_id="col_container"):
189
model = gr.Radio(
190
-
value="OpenAssistant/oasst-sft-(削除) 1 (削除ここまで)-pythia-12b",
191
choices=[
192
"OpenAssistant/oasst-sft-1-pythia-12b",
193
# "togethercomputer/GPT-NeoXT-Chat-Base-20B",
194
"google/flan-t5-xxl",
23
Returns:
24
(str, str, str, str): pre-prompt, username, bot name, separator
25
"""
26
+
if model (追記) in (追記ここまで) (追記) ( (追記ここまで)"OpenAssistant/oasst-sft-1-pythia-12b"(追記) , "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5") (追記ここまで):
27
return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
28
if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
29
return openchat_preprompt, "<human>: ", "<bot>: ", "\n"
65
66
partial_words = ""
67
68
+
if model (追記) in (追記ここまで) (追記) ( (追記ここまで)"OpenAssistant/oasst-sft-1-pythia-12b"(追記) , "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5") (追記ここまで):
69
iterator = client.generate_stream(
70
total_inputs,
71
typical_p=typical_p,
122
repetition_penalty,
123
watermark,
124
):
125
+
if value (追記) in (追記ここまで) (追記) ( (追記ここまで)"OpenAssistant/oasst-sft-1-pythia-12b"(追記) , "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5") (追記ここまで):
126
typical_p = typical_p.update(value=0.2, visible=True)
127
top_p = top_p.update(visible=False)
128
top_k = top_k.update(visible=False)
187
gr.Markdown(text_generation_inference, visible=True)
188
with gr.Column(elem_id="col_container"):
189
model = gr.Radio(
190
+
value="OpenAssistant/oasst-sft-(追記) 4 (追記ここまで)-pythia-12b(追記) -epoch-3.5 (追記ここまで)",
191
choices=[
192
+
"OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
193
"OpenAssistant/oasst-sft-1-pythia-12b",
194
# "togethercomputer/GPT-NeoXT-Chat-Base-20B",
195
"google/flan-t5-xxl",