Update models.py

models.py

@@ -22,41 +22,11 @@ DEFAULT_DST_INPUTS: Dict[str, str] = {
 
 
 DST_MODELS: Dict[str, Dict[str, Any]] = {
-    # "plt5-small": {
-    #     "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
-    #     "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-small-dst", use_auth_token=auth_token),
-    #     "default_input": DEFAULT_DST_INPUTS["polish"],
-    # },
-    # "plt5-base": {
-    #     "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
-    #     "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-dst", use_auth_token=auth_token),
-    #     "default_input": DEFAULT_DST_INPUTS["polish"],
-    # },
-    "plt5-base-poquad-dst-v2": {
-        "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
-        "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-base-poquad-dst-v2", use_auth_token=auth_token),
+    "plt5-large-poquad-dst-v2": {
+        "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/plt5-large-poquad-dst-v2", use_auth_token=auth_token),
+        "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/plt5-large-poquad-dst-v2", use_auth_token=auth_token),
         "default_input": DEFAULT_DST_INPUTS["polish"],
-    },
-    # "t5-small": {
-    #     "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
-    #     "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-small-dst", use_auth_token=auth_token),
-    #     "default_input": DEFAULT_DST_INPUTS["english"],
-    # },
-    # "t5-base": {
-    #     "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
-    #     "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/t5-base-dst", use_auth_token=auth_token),
-    #     "default_input": DEFAULT_DST_INPUTS["english"],
-    # },
-    # "flant5-small [EN/PL]": {
-    #     "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
-    #     "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-small-dst", use_auth_token=auth_token),
-    #     "default_input": DEFAULT_DST_INPUTS["english"],
-    # },
-    # "flant5-base [EN/PL]": {
-    #     "model": T5ForConditionalGeneration.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
-    #     "tokenizer": T5Tokenizer.from_pretrained("clarin-knext/flant5-base-dst", use_auth_token=auth_token),
-    #     "default_input": DEFAULT_DST_INPUTS["english"],
-    # },
+    }
 }