Skip to content

Commit 9b967e7

Browse files
authored
Add gpt neo X api doc (#2091)
1 parent 49f2c70 commit 9b967e7

File tree

1 file changed

+45
-0
lines changed

1 file changed

+45
-0
lines changed

scripts/hub_master.py

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1112,6 +1112,51 @@
11121112
],
11131113
},
11141114
],
1115+
},
1116+
{
1117+
"path": "gpt_neo_x/",
1118+
"title": "GPT-NeoX",
1119+
"toc": True,
1120+
"children": [
1121+
{
1122+
"path": "gpt_neo_x_tokenizer",
1123+
"title": "GPTNeoXTokenizer",
1124+
"generate": [
1125+
"keras_hub.tokenizers.GPTNeoXTokenizer",
1126+
"keras_hub.tokenizers.GPTNeoXTokenizer.from_preset",
1127+
],
1128+
},
1129+
{
1130+
"path": "gpt_neo_x_backbone",
1131+
"title": "GPTNeoXBackbone model",
1132+
"generate": [
1133+
"keras_hub.models.GPTNeoXBackbone",
1134+
"keras_hub.models.GPTNeoXBackbone.from_preset",
1135+
"keras_hub.models.GPTNeoXBackbone.token_embedding",
1136+
"keras_hub.models.GPTNeoXBackbone.enable_lora",
1137+
],
1138+
},
1139+
{
1140+
"path": "gpt_neo_x_causal_lm",
1141+
"title": "GPTNeoXCausalLM model",
1142+
"generate": [
1143+
"keras_hub.models.GPTNeoXCausalLM",
1144+
"keras_hub.models.GPTNeoXCausalLM.from_preset",
1145+
"keras_hub.models.GPTNeoXCausalLM.generate",
1146+
"keras_hub.models.GPTNeoXCausalLM.backbone",
1147+
"keras_hub.models.GPTNeoXCausalLM.preprocessor",
1148+
],
1149+
},
1150+
{
1151+
"path": "gpt_neo_x_causal_lm_preprocessor",
1152+
"title": "GPTNeoXCausalLMPreprocessor layer",
1153+
"generate": [
1154+
"keras_hub.models.GPTNeoXCausalLMPreprocessor",
1155+
"keras_hub.models.GPTNeoXCausalLMPreprocessor.from_preset",
1156+
"keras_hub.models.GPTNeoXCausalLMPreprocessor.tokenizer",
1157+
],
1158+
},
1159+
],
11151160
},
11161161
{
11171162
"path": "llama/",

0 commit comments

Comments
 (0)