Spaces:
Runtime error
Runtime error
OlivierDehaene
commited on
Commit
•
df2ff86
1
Parent(s):
f4b02db
remove joi
Browse files
app.py
CHANGED
@@ -13,8 +13,6 @@ openchat_preprompt = (
|
|
13 |
|
14 |
|
15 |
def get_client(model: str):
|
16 |
-
if model == "Rallio67/joi2_20Be_instruct_alpha":
|
17 |
-
return Client(os.getenv("JOI_API_URL"))
|
18 |
if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
|
19 |
return Client(os.getenv("OPENCHAT_API_URL"))
|
20 |
return InferenceAPIClient(model, token=os.getenv("HF_TOKEN", None))
|
@@ -27,8 +25,6 @@ def get_usernames(model: str):
|
|
27 |
"""
|
28 |
if model == "OpenAssistant/oasst-sft-1-pythia-12b":
|
29 |
return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
|
30 |
-
if model == "Rallio67/joi2_20Be_instruct_alpha":
|
31 |
-
return "", "User: ", "Joi: ", "\n\n"
|
32 |
if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
|
33 |
return openchat_preprompt, "<human>: ", "<bot>: ", "\n"
|
34 |
return "", "User: ", "Assistant: ", "\n"
|
@@ -190,7 +186,6 @@ with gr.Blocks(
|
|
190 |
choices=[
|
191 |
"OpenAssistant/oasst-sft-1-pythia-12b",
|
192 |
# "togethercomputer/GPT-NeoXT-Chat-Base-20B",
|
193 |
-
"Rallio67/joi2_20Be_instruct_alpha",
|
194 |
"google/flan-t5-xxl",
|
195 |
"google/flan-ul2",
|
196 |
"bigscience/bloom",
|
|
|
13 |
|
14 |
|
15 |
def get_client(model: str):
|
|
|
|
|
16 |
if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
|
17 |
return Client(os.getenv("OPENCHAT_API_URL"))
|
18 |
return InferenceAPIClient(model, token=os.getenv("HF_TOKEN", None))
|
|
|
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"
|
30 |
return "", "User: ", "Assistant: ", "\n"
|
|
|
186 |
choices=[
|
187 |
"OpenAssistant/oasst-sft-1-pythia-12b",
|
188 |
# "togethercomputer/GPT-NeoXT-Chat-Base-20B",
|
|
|
189 |
"google/flan-t5-xxl",
|
190 |
"google/flan-ul2",
|
191 |
"bigscience/bloom",
|