Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
•
e2bb507
1
Parent(s):
6c0544b
update
Browse files- utils/model.py +1 -1
utils/model.py
CHANGED
|
@@ -12,7 +12,7 @@ class Model(torch.nn.Module):
|
|
| 12 |
__model_list__ = [
|
| 13 |
"lmsys/vicuna-7b-v1.5",
|
| 14 |
"mistralai/Mistral-7B-Instruct-v0.1",
|
| 15 |
-
"meta-llama/Meta-Llama-3.1-8B"
|
| 16 |
]
|
| 17 |
|
| 18 |
def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
|
|
|
|
| 12 |
__model_list__ = [
|
| 13 |
"lmsys/vicuna-7b-v1.5",
|
| 14 |
"mistralai/Mistral-7B-Instruct-v0.1",
|
| 15 |
+
"meta-llama/Meta-Llama-3.1-8B-Instruct"
|
| 16 |
]
|
| 17 |
|
| 18 |
def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
|