cd4cfe7
1
2
3
4
5
6
7
from transformers import PretrainedConfig class Config(PretrainedConfig): model_type = "WGAN_GP" def __init__(self, **kwargs): super().__init__(**kwargs) self.cfg=kwargs
