diff --git a/src/models/asymmetric_encoder.py b/src/models/asymmetric_encoder.py index 1001c30..5e6d4ca 100644 --- a/src/models/asymmetric_encoder.py +++ b/src/models/asymmetric_encoder.py @@ -297,6 +297,7 @@ class AsymmetricEncoder(nn.Module): def __init__( self, + # !!! --------------------------------------------------------------- gate_type: ResidualGateType = ResidualGateType.simple_residual_one_gate, dino_web_path: str = "nn_models/DINO_WEB/dinov3-vitl16-pretrain-lvd1689m.pth", dino_sat_path: str = "nn_models/DINO_SAT/model.safetensors",