|
1112 | 1112 | ], |
1113 | 1113 | }, |
1114 | 1114 | ], |
| 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 | + ], |
1115 | 1160 | }, |
1116 | 1161 | { |
1117 | 1162 | "path": "llama/", |
|
0 commit comments