hmr2/models/__init__.py

from .smpl_wrapper import SMPL
from .hmr2 import HMR2
from .discriminator import Discriminator