third_part/ganimation_replicate/model/__init__.py

from .base_model import BaseModel
from .ganimation import GANimationModel
from .stargan import StarGANModel



def create_model(opt):
    # specify model name here
    if opt.model == "ganimation":
        instance = GANimationModel()
    elif opt.model == "stargan":
        instance = StarGANModel()
    else:
        instance = BaseModel()
    instance.initialize(opt)
    instance.setup()
    return instance