yuchenlin commited on
Commit
9374880
1 Parent(s): 1a5890e
Files changed (3) hide show
  1. __pycache__/utils.cpython-311.pyc +0 -0
  2. app.py +1 -1
  3. utils.py +2 -0
__pycache__/utils.cpython-311.pyc CHANGED
Binary files a/__pycache__/utils.cpython-311.pyc and b/__pycache__/utils.cpython-311.pyc differ
 
app.py CHANGED
@@ -71,7 +71,7 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label) as demo:
71
  with gr.Column():
72
  gr.Markdown(HEADER_MD)
73
  model_name = gr.Radio(["Llama-3-8B", "Llama-3-70B", "Mistral-7B-v0.1",
74
- "Mixtral-8x22B", "Yi-6B", "Yi-34B", "Llama-2-7B", "Llama-2-70B", "OLMO"]
75
  , value="Llama-3-8B", label="Base LLM name")
76
  with gr.Column():
77
  together_api_key = gr.Textbox(label="🔑 Together APIKey", placeholder="Enter your Together API Key. Leave it blank to use our key with limited usage.", type="password", elem_id="api_key")
 
71
  with gr.Column():
72
  gr.Markdown(HEADER_MD)
73
  model_name = gr.Radio(["Llama-3-8B", "Llama-3-70B", "Mistral-7B-v0.1",
74
+ "Mixtral-8x22B", "Qwen1.5-72B", "Yi-34B", "Llama-2-7B", "Llama-2-70B", "OLMO"]
75
  , value="Llama-3-8B", label="Base LLM name")
76
  with gr.Column():
77
  together_api_key = gr.Textbox(label="🔑 Together APIKey", placeholder="Enter your Together API Key. Leave it blank to use our key with limited usage.", type="password", elem_id="api_key")
utils.py CHANGED
@@ -28,6 +28,8 @@ def model_name_mapping(model_name):
28
  _model_name = "zero-one-ai/Yi-6B"
29
  elif model_name == "OLMO":
30
  _model_name = "allenai/OLMo-7B"
 
 
31
  else:
32
  raise ValueError("Invalid model name")
33
  return _model_name
 
28
  _model_name = "zero-one-ai/Yi-6B"
29
  elif model_name == "OLMO":
30
  _model_name = "allenai/OLMo-7B"
31
+ elif model_name == "Qwen1.5-72B":
32
+ _model_name = "Qwen/Qwen1.5-72B"
33
  else:
34
  raise ValueError("Invalid model name")
35
  return _model_name