Safetensors
custom_code
File size: 252 Bytes
4333430
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from transformers import Dinov2Model


def build_vision_encoder(config):
    if config.model_type == "dinov2":
        model = Dinov2Model.from_pretrained(config.pretrained_name_or_path)

    else:
        raise NotImplementedError()

    return model