Update models.py

models.py

@@ -41,6 +41,7 @@ def return_models(task_name):
             "together_computer",
             "mosaic_ml",
             "h20ai",
+            "deciai"
         )
     return models_parent
 
@@ -315,6 +316,7 @@ def return_text2text_generation_models(model_parent, count=False):
         "microsoft/xprophetnet-large-wiki100-cased-xglue-ntg",
         "microsoft/prophetnet-large-uncased-squad-qg",
         "microsoft/prophetnet-large-uncased-cnndm",
+        
     )
     starmpcc_models_list = (
         "starmpcc/Asclepius-13B",
@@ -429,7 +431,7 @@ def return_text2text_generation_models(model_parent, count=False):
         "langboat": langboat_models_list,
         "haining": haining_models_list,
         "kaludi": kaludi_models_list,
-        "allenai": allenai_models_list
+        "allenai": allenai_models_list,
         }
     if count is True:
         models_count = 0
@@ -799,6 +801,8 @@ def return_text_generation_models(model_parent, count=False):
         "microsoft/CodeGPT-small-py-adaptedGPT2",
         "microsoft/CodeGPT-small-java",
         "microsoft/CodeGPT-small-java-adaptedGPT2",
+        "microsoft/phi-1",
+        "microsoft/phi-1_5",
     )
     
     langboat_models_list = (
@@ -851,6 +855,12 @@ def return_text_generation_models(model_parent, count=False):
         'allenai/open-instruct-human-mix-65b'
         )
 
+    deciai_models_list = (
+        "Deci/DeciLM-6b",
+        "Deci/DeciLM-6b-instruct",
+        "Deci/DeciCoder-1b",
+    )
+
     model_dict = {
         "google": google_models_list,
         "mbzuai": mbzuai_models_list,
@@ -872,7 +882,8 @@ def return_text_generation_models(model_parent, count=False):
         "facebook": facebook_models_list,
         "microsoft": microsoft_models_list,
         "langboat": langboat_models_list,
-        "allenai": allenai_models_list
+        "allenai": allenai_models_list,
+        "deciai": deciai_models_list
         }
     
     if count is True: