# Copyright (c) Alibaba, Inc. and its affiliates.

from .modules.quadtree_attention import QTAttA, QTAttB