Sharathhebbar24 commited on
Commit
d0af6a4
1 Parent(s): fc4ae9c

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +13 -2
models.py CHANGED
@@ -41,6 +41,7 @@ def return_models(task_name):
41
  "together_computer",
42
  "mosaic_ml",
43
  "h20ai",
 
44
  )
45
  return models_parent
46
 
@@ -315,6 +316,7 @@ def return_text2text_generation_models(model_parent, count=False):
315
  "microsoft/xprophetnet-large-wiki100-cased-xglue-ntg",
316
  "microsoft/prophetnet-large-uncased-squad-qg",
317
  "microsoft/prophetnet-large-uncased-cnndm",
 
318
  )
319
  starmpcc_models_list = (
320
  "starmpcc/Asclepius-13B",
@@ -429,7 +431,7 @@ def return_text2text_generation_models(model_parent, count=False):
429
  "langboat": langboat_models_list,
430
  "haining": haining_models_list,
431
  "kaludi": kaludi_models_list,
432
- "allenai": allenai_models_list
433
  }
434
  if count is True:
435
  models_count = 0
@@ -799,6 +801,8 @@ def return_text_generation_models(model_parent, count=False):
799
  "microsoft/CodeGPT-small-py-adaptedGPT2",
800
  "microsoft/CodeGPT-small-java",
801
  "microsoft/CodeGPT-small-java-adaptedGPT2",
 
 
802
  )
803
 
804
  langboat_models_list = (
@@ -851,6 +855,12 @@ def return_text_generation_models(model_parent, count=False):
851
  'allenai/open-instruct-human-mix-65b'
852
  )
853
 
 
 
 
 
 
 
854
  model_dict = {
855
  "google": google_models_list,
856
  "mbzuai": mbzuai_models_list,
@@ -872,7 +882,8 @@ def return_text_generation_models(model_parent, count=False):
872
  "facebook": facebook_models_list,
873
  "microsoft": microsoft_models_list,
874
  "langboat": langboat_models_list,
875
- "allenai": allenai_models_list
 
876
  }
877
 
878
  if count is True:
 
41
  "together_computer",
42
  "mosaic_ml",
43
  "h20ai",
44
+ "deciai"
45
  )
46
  return models_parent
47
 
 
316
  "microsoft/xprophetnet-large-wiki100-cased-xglue-ntg",
317
  "microsoft/prophetnet-large-uncased-squad-qg",
318
  "microsoft/prophetnet-large-uncased-cnndm",
319
+
320
  )
321
  starmpcc_models_list = (
322
  "starmpcc/Asclepius-13B",
 
431
  "langboat": langboat_models_list,
432
  "haining": haining_models_list,
433
  "kaludi": kaludi_models_list,
434
+ "allenai": allenai_models_list,
435
  }
436
  if count is True:
437
  models_count = 0
 
801
  "microsoft/CodeGPT-small-py-adaptedGPT2",
802
  "microsoft/CodeGPT-small-java",
803
  "microsoft/CodeGPT-small-java-adaptedGPT2",
804
+ "microsoft/phi-1",
805
+ "microsoft/phi-1_5",
806
  )
807
 
808
  langboat_models_list = (
 
855
  'allenai/open-instruct-human-mix-65b'
856
  )
857
 
858
+ deciai_models_list = (
859
+ "Deci/DeciLM-6b",
860
+ "Deci/DeciLM-6b-instruct",
861
+ "Deci/DeciCoder-1b",
862
+ )
863
+
864
  model_dict = {
865
  "google": google_models_list,
866
  "mbzuai": mbzuai_models_list,
 
882
  "facebook": facebook_models_list,
883
  "microsoft": microsoft_models_list,
884
  "langboat": langboat_models_list,
885
+ "allenai": allenai_models_list,
886
+ "deciai": deciai_models_list
887
  }
888
 
889
  if count is True: