Skip to content

Commit f1b99de

Browse files
add ESM API usage details
1 parent 52127e7 commit f1b99de

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
@@ -825,6 +825,68 @@
825825
},
826826
],
827827
},
828+
{
829+
"path": "esm/",
830+
"title": "ESM",
831+
"toc": True,
832+
"children": [
833+
{
834+
"path": "esm_tokenizer",
835+
"title": "ESMTokenizer",
836+
"generate": [
837+
"keras_hub.tokenizers.ESMTokenizer",
838+
"keras_hub.tokenizers.ESMTokenizer.from_preset",
839+
],
840+
},
841+
{
842+
"path": "esm_backbone",
843+
"title": "ESMBackbone model",
844+
"generate": [
845+
"keras_hub.models.ESMBackbone",
846+
"keras_hub.models.ESMBackbone.from_preset",
847+
"keras_hub.models.ESMBackbone.token_embedding",
848+
],
849+
},
850+
{
851+
"path": "esm_classifier",
852+
"title": "ESMProteinClassifier model",
853+
"generate": [
854+
"keras_hub.models.ESMProteinClassifier",
855+
"keras_hub.models.ESMProteinClassifier.from_preset",
856+
"keras_hub.models.ESMProteinClassifier.backbone",
857+
"keras_hub.models.ESMProteinClassifier.preprocessor",
858+
],
859+
},
860+
{
861+
"path": "esm_classifier_preprocessor",
862+
"title": "ESMProteinClassifierPreprocessor layer",
863+
"generate": [
864+
"keras_hub.models.ESMProteinClassifierPreprocessor",
865+
"keras_hub.models.ESMProteinClassifierPreprocessor.from_preset",
866+
"keras_hub.models.ESMProteinClassifierPreprocessor.tokenizer",
867+
],
868+
},
869+
{
870+
"path": "esm_masked_plm",
871+
"title": "ESMMaskedPLM model",
872+
"generate": [
873+
"keras_hub.models.ESMMaskedPLM",
874+
"keras_hub.models.ESMMaskedPLM.from_preset",
875+
"keras_hub.models.ESMMaskedPLM.backbone",
876+
"keras_hub.models.ESMMaskedPLM.preprocessor",
877+
],
878+
},
879+
{
880+
"path": "esm_masked_plm_preprocessor",
881+
"title": "ESMMaskedPLMPreprocessor layer",
882+
"generate": [
883+
"keras_hub.models.ESMMaskedPLMPreprocessor",
884+
"keras_hub.models.ESMMaskedPLMPreprocessor.from_preset",
885+
"keras_hub.models.ESMMaskedPLMPreprocessor.tokenizer",
886+
],
887+
},
888+
],
889+
},
828890
{
829891
"path": "falcon/",
830892
"title": "Falcon",

0 commit comments

Comments
 (0)