File size: 360 Bytes
c546054 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import PretrainedConfig
class StormOracleConfig(PretrainedConfig):
model_type = "storm_oracle"
def __init__(self,
in_channels: int = 3,
image_size: int = 256,
**kwargs):
super().__init__(**kwargs)
self.in_channels = in_channels
self.image_size = image_size
|