__init__.py

from .configuration_gpt_and_prejudice import GptAndPrejudiceConfig
from .modeling_gpt_and_prejudice import (
    GptAndPrejudiceForCausalLM,
    GptAndPrejudiceCausalLMOutput,
)