import gin

@gin.configurable
class DINOv3ModelsConfig:
    def __init__(
        self,
        dino_web_path: str = "nn_models/DINO_WEB/dinov3-vitl16-pretrain-lvd1689m.pth",
        dino_sat_path: str = "nn_models/DINO_SAT/model.safetensors",
        shared_encoder: bool = True,
        mona_bottleneck: int = 64,
        mona_last_n_blocks: int = 12,
    ) -> None:
        self.dino_web_path = dino_web_path
        self.dino_sat_path = dino_sat_path
        self.shared_encoder = shared_encoder
        self.mona_bottleneck = mona_bottleneck
        self.mona_last_n_blocks = mona_last_n_blocks

        
