From 40a4f6daa96425e57d2f6f90fb1e99c099f5e40e Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Mon, 12 Aug 2024 22:42:57 +0000 Subject: [PATCH] import layer fix --- keras_cv/src/layers/vit_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_cv/src/layers/vit_layers.py b/keras_cv/src/layers/vit_layers.py index fddf731ef1..3e7e8087c9 100644 --- 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.