Update vit_layers.py

keras-team · Aug 12, 2024 · 708541a · 708541a
1 parent e7bd438
commit 708541a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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")