From da5922d0a5524887aa492f83153291590d192d5c Mon Sep 17 00:00:00 2001
From: Joan Puigcerver <jpuigcerver@google.com>
Date: Mon, 29 Apr 2024 08:55:48 -0700
Subject: [PATCH] Internal changes.

PiperOrigin-RevId: 629082778
---
 vmoe/nn/vit_moe.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/vmoe/nn/vit_moe.py b/vmoe/nn/vit_moe.py
index d9f0277..7319db4 100644
--- a/vmoe/nn/vit_moe.py
+++ b/vmoe/nn/vit_moe.py
@@ -34,7 +34,6 @@
 Shape = Iterable[int]
 
 
-
 # Slight modification of the VisionTransformer's MlpBlock API.
 class MlpBlock(models_vit.MlpBlock):
   dtype: Optional[DType] = None