File size: 283 Bytes
0c6516e |
1 2 3 4 5 6 7 8 9 |
from transformers import PretrainedConfig
class ISNetConfig(PretrainedConfig):
def __init__(self, in_channels: int = 3, out_channels: int = 1, **kwargs) -> None:
super().__init__(**kwargs)
self.in_channels = in_channels
self.out_channels = out_channels
|