Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
•
f654526
1
Parent(s):
4fb58cc
delete falcon and add llama3
Browse files- utils/model.py +1 -1
utils/model.py
CHANGED
@@ -6,7 +6,7 @@ class Model(torch.nn.Module):
|
|
6 |
number_of_models = 0
|
7 |
__model_list__ = [
|
8 |
"lmsys/vicuna-7b-v1.5",
|
9 |
-
"
|
10 |
]
|
11 |
|
12 |
def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
|
|
|
6 |
number_of_models = 0
|
7 |
__model_list__ = [
|
8 |
"lmsys/vicuna-7b-v1.5",
|
9 |
+
"meta-llama/Meta-Llama-3.1-8B"
|
10 |
]
|
11 |
|
12 |
def __init__(self, model_name="lmsys/vicuna-7b-v1.5") -> None:
|