kz209 commited on
Commit
f654526
1 Parent(s): 4fb58cc

delete falcon and add llama3

Browse files
Files changed (1) hide show
  1. 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
- "tiiuae/falcon-7b-instruct"
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: