Skip to content

Commit e8dea82

Browse files
Wauplincoyotte508
andauthored
Add support for KerasNLP library (#616)
Following recent work to integrate KerasNLP library with the Hub, let's add it in our libraries list. KerasNLP is based on Keras3 but for now let's focus on the KerasNLP side of things (we also have on-going work for Keras but that's orthogonal). This PR will add a pretty name for "KerasNLP" tag + add urls to official docs and github repo. We will be able to generate code snippet in the future as well but [it should be clarified](keras-team/keras-hub#1529 (comment)) first. I suggest that we merge this PR "as-is" and add support for code snippets later when it's ready. About download counts, KerasNLP uses a `config.json` file so it should already count correctly. Bonus: first KerasNLP models on the Hub? https://huggingface.co/google/gemma-2b-it-keras/discussions/2. Model can be loaded with: ```py import keras_nlp gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("hf://google/gemma-2b-it-keras") ``` :tada: Co-authored-by: Eliott C <[email protected]>
1 parent 1f49e2a commit e8dea82

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
185185
filter: true,
186186
countDownloads: { term: { path: "saved_model.pb" } },
187187
},
188+
"keras-nlp": {
189+
prettyLabel: "KerasNLP",
190+
repoName: "KerasNLP",
191+
repoUrl: "https://keras.io/keras_nlp/",
192+
docsUrl: "https://github.com/keras-team/keras-nlp",
193+
},
188194
k2: {
189195
prettyLabel: "K2",
190196
repoName: "k2",

0 commit comments

Comments
 (0)