Fix RotaryEmbedding import (#1217)

* fix import * format
keras-team · Aug 23, 2023 · 66572eb · 66572eb
1 parent 2d187c1
commit 66572eb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras_nlp/layers/__init__.py b/keras_nlp/layers/__init__.py
@@ -18,6 +18,7 @@
 from keras_nlp.layers.modeling.f_net_encoder import FNetEncoder
 from keras_nlp.layers.modeling.masked_lm_head import MaskedLMHead
 from keras_nlp.layers.modeling.position_embedding import PositionEmbedding
+from keras_nlp.layers.modeling.rotary_embedding import RotaryEmbedding
 from keras_nlp.layers.modeling.sine_position_encoding import (
     SinePositionEncoding,
 )