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) |