Hugging Face's logo

Spaces:
open-nlp
/
Chris-lab
Runtime error

App Files Community
2
kz209 commited on
Commit
c9c9f16
·
1 Parent(s): d5f9bcf

add llama3

Browse files
Files changed (1) hide show
  1. utils/model.py +2 -1
utils/model.py CHANGED
@@ -11,7 +11,8 @@ class Model(torch.nn.Module):
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:
 
11
  number_of_models = 0
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: