Update more supported ESM models

huggingface · Dec 10, 2023 · 32752c6 · 32752c6
1 parent 775815a
commit 32752c6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/scripts/supported_models.py b/scripts/supported_models.py
@@ -358,6 +358,16 @@
             'facebook/esm2_t30_150M_UR50D',
             'facebook/esm2_t33_650M_UR50D',
         ],
+
+        # Token classification
+        'token-classification': [
+            'AmelieSchreiber/esm2_t6_8M_UR50D_rna_binding_site_predictor',
+        ],
+
+        # Zero-shot classification
+        'zero-shot-classification': [
+            'AmelieSchreiber/esm2_t6_8M_UR50D_sequence_classifier_v1',
+        ],
     },
     'falcon': {
         # Text generation