import layer fix

keras-team · Aug 12, 2024 · 40a4f6d · 40a4f6d
1 parent 7440b65
commit 40a4f6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_cv/src/layers/vit_layers.py b/keras_cv/src/layers/vit_layers.py
@@ -225,7 +225,7 @@ def get_config(self):
 
 
 @keras_cv_export("keras_cv.layers.Unpatching")
-class Unpatching(layers.Layer):
+class Unpatching(keras.layers.Layer):
     """
     Layer to unpatchify image data.