# __init__.py in the model repo
from .custom_modeling import SafeGenerationModel