kz209 commited on
Commit
3f40f6e
1 Parent(s): 9779cd8

add t5-large

Browse files
Files changed (1) hide show
  1. utils/model.py +1 -0
utils/model.py CHANGED
@@ -13,6 +13,7 @@ class Model(torch.nn.Module):
13
  number_of_models = 0
14
  __model_list__ = [
15
  "lmsys/vicuna-7b-v1.5",
 
16
  "mistralai/Mistral-7B-Instruct-v0.1",
17
  "meta-llama/Meta-Llama-3.1-8B-Instruct"
18
  ]
 
13
  number_of_models = 0
14
  __model_list__ = [
15
  "lmsys/vicuna-7b-v1.5",
16
+ "google-t5/t5-large",
17
  "mistralai/Mistral-7B-Instruct-v0.1",
18
  "meta-llama/Meta-Llama-3.1-8B-Instruct"
19
  ]