<filename>vformer/decoder/__init__.py
from .mlp import MLPDecoder
from .perceiver_io import PerceiverIODecoder
from .task_heads import *

<filename>vformer/decoder/__init__.py
from .mlp import MLPDecoder
from .perceiver_io import PerceiverIODecoder
from .task_heads import *
