| from transformers import PretrainedConfig | |
| class RealESRGANConfig(PretrainedConfig): | |
| model_type = "realesrgan" | |
| def __init__( | |
| self, | |
| num_in_ch: int = 3, | |
| num_out_ch: int = 3, | |
| num_feat: int = 64, | |
| num_block: int = 23, | |
| num_grow_ch: int = 32, | |
| scale: int = 4, | |
| **kwargs, | |
| ): | |
| self.num_in_ch = num_in_ch | |
| self.num_out_ch = num_out_ch | |
| self.num_feat = num_feat | |
| self.num_block = num_block | |
| self.num_grow_ch = num_grow_ch | |
| self.scale = scale | |
| super().__init__(**kwargs) | |