dropout layer missing as per official implementation

keras_hub/src/models/vit/vit_layers.py

@@ -65,6 +65,7 @@ def build(self, input_shape):
 
     def call(self, inputs):
         x = self.dense_1(inputs)
+        x = self.dropout(x)
         x = self.dense_2(x)
         out = self.dropout(x)
         return out