Update vit_layers.py

keras-team · Aug 12, 2024 · 8d9efc4 · 8d9efc4
1 parent cf33ec8
commit 8d9efc4
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
@@ -15,7 +15,7 @@
 import math
 
 import tensorflow as tf
-from keras_cv.src.backend import layers
+from keras_cv.src.backend.keras import layers
 from keras_cv.src.backend import ops
 
 from keras_cv.src.api_export import keras_cv_export