Spaces:
Runtime error
Runtime error
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:
|