config.py

from transformers import PretrainedConfig
class Config(PretrainedConfig):
    model_type = "WGAN_GP"
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.cfg=kwargs