Skip to content

Commit 9bca37d

Browse files
add ESM API usage details (#2163)
1 parent e330ef2 commit 9bca37d

File tree

1 file changed

+62
-0
lines changed

1 file changed

+62
-0
lines changed

scripts/hub_master.py

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -848,6 +848,68 @@
848848
},
849849
],
850850
},
851+
{
852+
"path": "esm/",
853+
"title": "ESM",
854+
"toc": True,
855+
"children": [
856+
{
857+
"path": "esm_tokenizer",
858+
"title": "ESMTokenizer",
859+
"generate": [
860+
"keras_hub.tokenizers.ESMTokenizer",
861+
"keras_hub.tokenizers.ESMTokenizer.from_preset",
862+
],
863+
},
864+
{
865+
"path": "esm_backbone",
866+
"title": "ESMBackbone model",
867+
"generate": [
868+
"keras_hub.models.ESMBackbone",
869+
"keras_hub.models.ESMBackbone.from_preset",
870+
"keras_hub.models.ESMBackbone.token_embedding",
871+
],
872+
},
873+
{
874+
"path": "esm_classifier",
875+
"title": "ESMProteinClassifier model",
876+
"generate": [
877+
"keras_hub.models.ESMProteinClassifier",
878+
"keras_hub.models.ESMProteinClassifier.from_preset",
879+
"keras_hub.models.ESMProteinClassifier.backbone",
880+
"keras_hub.models.ESMProteinClassifier.preprocessor",
881+
],
882+
},
883+
{
884+
"path": "esm_classifier_preprocessor",
885+
"title": "ESMProteinClassifierPreprocessor layer",
886+
"generate": [
887+
"keras_hub.models.ESMProteinClassifierPreprocessor",
888+
"keras_hub.models.ESMProteinClassifierPreprocessor.from_preset",
889+
"keras_hub.models.ESMProteinClassifierPreprocessor.tokenizer",
890+
],
891+
},
892+
{
893+
"path": "esm_masked_plm",
894+
"title": "ESMMaskedPLM model",
895+
"generate": [
896+
"keras_hub.models.ESMMaskedPLM",
897+
"keras_hub.models.ESMMaskedPLM.from_preset",
898+
"keras_hub.models.ESMMaskedPLM.backbone",
899+
"keras_hub.models.ESMMaskedPLM.preprocessor",
900+
],
901+
},
902+
{
903+
"path": "esm_masked_plm_preprocessor",
904+
"title": "ESMMaskedPLMPreprocessor layer",
905+
"generate": [
906+
"keras_hub.models.ESMMaskedPLMPreprocessor",
907+
"keras_hub.models.ESMMaskedPLMPreprocessor.from_preset",
908+
"keras_hub.models.ESMMaskedPLMPreprocessor.tokenizer",
909+
],
910+
},
911+
],
912+
},
851913
{
852914
"path": "falcon/",
853915
"title": "Falcon",

0 commit comments

Comments
 (0)