diff --git a/keras_cv/models/segmentation/segformer/segformer_presets.py b/keras_cv/models/segmentation/segformer/segformer_presets.py index 02068c01df..15d2bac8fe 100644 --- a/keras_cv/models/segmentation/segformer/segformer_presets.py +++ b/keras_cv/models/segmentation/segformer/segformer_presets.py @@ -13,6 +13,9 @@ # limitations under the License. """SegFormer model preset configurations.""" +from keras_cv.models.backbones.mix_transformer.mix_transformer_backbone_presets import ( # noqa: E501 + backbone_presets, +) presets_no_weights = { "segformer_b0": {