OlivierDehaene commited on
Commit
df2ff86
1 Parent(s): f4b02db

remove joi

Browse files
Files changed (1) hide show
  1. app.py +0 -5
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",