Update segformer_presets with backbone import (fix merge) (keras-team…

…#2225) * Update segformer_presets with backbone import * Update format.
nkovela1 · Dec 11, 2023 · 71245a8 · 71245a8
1 parent 6ea822d
commit 71245a8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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": {