models/__init__.py

from .kts_src import *
from .clip_model import *
from .grit_model import *