File size: 584 Bytes
41d73ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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)
|