From 708541aebfd292f5939b93ac099e8e1b81564e84 Mon Sep 17 00:00:00 2001 From: Divyashree Sreepathihalli Date: Mon, 12 Aug 2024 15:14:21 -0700 Subject: [PATCH] Update vit_layers.py --- keras_cv/src/layers/vit_layers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/keras_cv/src/layers/vit_layers.py b/keras_cv/src/layers/vit_layers.py index 5b0a4c4d3d..b9693a27d1 100644 --- a/keras_cv/src/layers/vit_layers.py +++ b/keras_cv/src/layers/vit_layers.py @@ -15,10 +15,10 @@ import math import tensorflow as tf -from keras_cv.backend import layers -from keras_cv.backend import ops +from keras_cv.src.backend import layers +from keras_cv.src.backend import ops -from keras_cv.api_export import keras_cv_export +from keras_cv.src.api_export import keras_cv_export @keras_cv_export("keras_cv.layers.PatchingAndEmbedding")