configuration_doduo.py

from transformers import PretrainedConfig
from typing import List

class DoduoConfig(PretrainedConfig):
    model_type = "doduo"

    def __init__(
        self,
        dino_corr_mask_ratio: float = 0.99,
        **kwargs,
    ):
        self.dino_corr_mask_ratio = dino_corr_mask_ratio
        super().__init__(**kwargs)