Spaces:
Running
Running
add qweb
Browse files- __pycache__/utils.cpython-311.pyc +0 -0
- app.py +1 -1
- 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", "
|
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
|