Unverified Commit ba4b0db5 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Add missing `_version` to the MLPBlock (#6113)

* Add missing `_version` to the MLPBlock

* fix linter
parent 2a810713
......@@ -40,6 +40,8 @@ class ConvStemConfig(NamedTuple):
class MLPBlock(MLP):
"""Transformer MLP block."""
_version = 2
def __init__(self, in_dim: int, mlp_dim: int, dropout: float):
super().__init__(in_dim, [mlp_dim, in_dim], activation_layer=nn.GELU, inplace=None, dropout=dropout)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment