ola_vlm/model/aux_heads/__init__.py

from .da_v2_head import DepthHead, DAv2_Head, DepthProbeHead, TaskTokenDepthHead
from .oneformer_head import OneFormerSegHead, OneFormerTaskTokenSegHead
from .gen_head import GenHead, TaskTokenGenHead