Skip to content

Commit 69b8cdf

Browse files
committed
Add example for backbone
1 parent a548a1c commit 69b8cdf

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

keras_nlp/models/bloom/bloom_backbone.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,10 @@ class BloomBackbone(Backbone):
6363
"padding_mask": np.array([[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0]]),
6464
}
6565
66+
# Pretrained BLOOM decoder.
67+
model = keras_nlp.models.BloomBackbone.from_preset("bloom_560m_multi")
68+
model(input_data)
69+
6670
# Randomly initialized BLOOM decoder with a custom config.
6771
model = keras_nlp.models.BloomBackbone(
6872
vocabulary_size=10,

0 commit comments

Comments
 (0)