Fix build errors with new keras-core version

mattdangerw · Sep 18, 2023 · 0f7cd8c · 0f7cd8c
1 parent 2707f90
commit 0f7cd8c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/keras_cv/layers/preprocessing/base_image_augmentation_layer.py b/keras_cv/layers/preprocessing/base_image_augmentation_layer.py
@@ -135,6 +135,7 @@ def __init__(self, seed=None, **kwargs):
             seed=seed, force_generator=force_generator
         )
         super().__init__(**kwargs)
+        self.built = True
         self._convert_input_args = False
 
     @property

diff --git a/keras_cv/models/backbones/csp_darknet/csp_darknet_utils.py b/keras_cv/models/backbones/csp_darknet/csp_darknet_utils.py
@@ -310,7 +310,8 @@ def __init__(
                 )
             )
 
-        self.add = keras.layers.Add()
+        if self.residual:
+            self.add = keras.layers.Add()
         self.concatenate = keras.layers.Concatenate()
 
         self.darknet_conv3 = DarknetConvBlock(