4
4
[ ![ contributions welcome] ( https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat )] ( https://github.com/keras-team/keras-hub/issues )
5
5
6
6
> [ !IMPORTANT]
7
- > 📢 KerasNLP is becoming KerasHub! 📢 Read
7
+ > 📢 KerasNLP is now KerasHub! 📢 Read
8
8
> [ the announcement] ( https://github.com/keras-team/keras-hub/issues/1831 ) .
9
9
>
10
10
> We have renamed the repo to KerasHub in preparation for the release, but have not yet
@@ -26,7 +26,7 @@ All models support JAX, TensorFlow, and PyTorch from a single model
26
26
definition and can be fine-tuned on GPUs and TPUs out of the box. Models can
27
27
be trained on individual accelerators with built-in PEFT techniques, or
28
28
fine-tuned at scale with model and data parallel training. See our
29
- [ Getting Started guide] ( https://keras.io/guides/keras_nlp /getting_started )
29
+ [ Getting Started guide] ( https://keras.io/guides/keras_hub /getting_started )
30
30
to start learning our API. Browse our models on
31
31
[ Kaggle] ( https://www.kaggle.com/organizations/keras/models ) .
32
32
We welcome contributions.
@@ -35,9 +35,9 @@ We welcome contributions.
35
35
36
36
### For everyone
37
37
38
- - [ Home Page] ( https://keras.io/keras_nlp )
39
- - [ Developer Guides] ( https://keras.io/guides/keras_nlp )
40
- - [ API Reference] ( https://keras.io/api/keras_nlp )
38
+ - [ Home Page] ( https://keras.io/keras_hub )
39
+ - [ Developer Guides] ( https://keras.io/guides/keras_hub )
40
+ - [ API Reference] ( https://keras.io/api/keras_hub )
41
41
- [ Pre-trained Models] ( https://www.kaggle.com/organizations/keras/models )
42
42
43
43
### For contributors
@@ -56,7 +56,7 @@ Fine-tune a BERT classifier on IMDb movie reviews:
56
56
import os
57
57
os.environ[" KERAS_BACKEND" ] = " jax" # Or "tensorflow" or "torch"!
58
58
59
- import keras_nlp
59
+ import keras_hub
60
60
import tensorflow_datasets as tfds
61
61
62
62
imdb_train, imdb_test = tfds.load(
@@ -67,7 +67,7 @@ imdb_train, imdb_test = tfds.load(
67
67
)
68
68
69
69
# Load a BERT model.
70
- classifier = keras_nlp .models.Classifier.from_preset(
70
+ classifier = keras_hub .models.Classifier.from_preset(
71
71
" bert_base_en" ,
72
72
num_classes = 2 ,
73
73
activation = " softmax" ,
@@ -79,25 +79,17 @@ classifier.fit(imdb_train, validation_data=imdb_test)
79
79
classifier.predict([" What an amazing movie!" , " A total waste of my time." ])
80
80
```
81
81
82
- Try it out [ in a colab] ( https://colab.research.google.com/gist/mattdangerw/e457e42d5ea827110c8d5cb4eb9d9a07/kerasnlp-quickstart.ipynb ) .
82
+ Try it out [ in a colab] ( https://colab.research.google.com/drive/1gSWkh3yOLwmKAaNh2dQQ6kQIlnGte7P2?usp=sharing ) .
83
83
For more in depth guides and examples, visit
84
- [ keras.io/keras_nlp ] ( https://keras.io/keras_nlp / ) .
84
+ [ keras.io/keras_hub ] ( https://keras.io/keras_hub / ) .
85
85
86
86
## Installation
87
87
88
- KerasHub is currently in pre-release. Note that pre-release versions may
89
- introduce breaking changes to the API in future versions. For a stable and
90
- supported experience, we recommend installing ` keras-nlp ` version 0.15.1:
91
-
92
- ``` bash
93
- pip install keras-nlp==0.15.1
94
- ```
95
-
96
- To try out the latest pre-release version of KerasHub, you can use
88
+ To try out the latest version of KerasHub, you can use
97
89
our nightly package:
98
90
99
91
``` bash
100
- pip install keras-hub-nightly
92
+ pip install keras-hub
101
93
```
102
94
103
95
KerasHub currently requires TensorFlow to be installed for use of the
0 commit comments