ControlNet/marlin_vit_base_ytf

1from transformers import PretrainedConfig2 3 4class MarlinConfig(PretrainedConfig):5    model_type = "marlin"6 7    def __init__(self, **kwargs):8        self.img_size = kwargs.pop("img_size", None)9        self.patch_size = kwargs.pop("patch_size", None)10        self.n_frames = kwargs.pop("n_frames", None)11        self.encoder_embed_dim = kwargs.pop("encoder_embed_dim", None)12        self.encoder_depth = kwargs.pop("encoder_depth", None)13        self.encoder_num_heads = kwargs.pop("encoder_num_heads", None)14        self.decoder_embed_dim = kwargs.pop("decoder_embed_dim", None)15        self.decoder_depth = kwargs.pop("decoder_depth", None)16        self.decoder_num_heads = kwargs.pop("decoder_num_heads", None)17        self.mlp_ratio = kwargs.pop("mlp_ratio", None)18        self.qkv_bias = kwargs.pop("qkv_bias", None)19        self.qk_scale = kwargs.pop("qk_scale", None)20        self.drop_rate = kwargs.pop("drop_rate", None)21        self.attn_drop_rate = kwargs.pop("attn_drop_rate", None)22        self.norm_layer = kwargs.pop("norm_layer", None)23        self.init_values = kwargs.pop("init_values", None)24        self.tubelet_size = kwargs.pop("tubelet_size", None)25        self.as_feature_extractor = kwargs.pop("as_feature_extractor", True)26 27        super().__init__(**kwargs)28