Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
·
c689aee
1
Parent(s):
e63d745
change to mistral
Browse files- utils/model.py +1 -1
utils/model.py
CHANGED
@@ -11,7 +11,7 @@ class Model(torch.nn.Module):
|
|
11 |
number_of_models = 0
|
12 |
__model_list__ = [
|
13 |
"lmsys/vicuna-7b-v1.5",
|
14 |
-
"
|
15 |
]
|
16 |
|
17 |
def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
|
|
|
11 |
number_of_models = 0
|
12 |
__model_list__ = [
|
13 |
"lmsys/vicuna-7b-v1.5",
|
14 |
+
"mistralai/Mistral-7B-Instruct-v0.1"
|
15 |
]
|
16 |
|
17 |
def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
|