Skip to content

Commit

Permalink
Internal changes.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 629082778
  • Loading branch information
jpuigcerver authored and copybara-github committed Apr 29, 2024
1 parent c65605b commit da5922d
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion vmoe/nn/vit_moe.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@
Shape = Iterable[int]



# Slight modification of the VisionTransformer's MlpBlock API.
class MlpBlock(models_vit.MlpBlock):
dtype: Optional[DType] = None
Expand Down

0 comments on commit da5922d

Please sign in to comment.