"""Sky CREST Configuration — 0labs"""
from transformers import PretrainedConfig

class SkyCRESTConfig(PretrainedConfig):
    model_type = "sky-crest"

    def __init__(self, crest_max_steps=4, **kwargs):
        self.crest_max_steps = crest_max_steps
        super().__init__(**kwargs)