from src.model.utransformer import UTransformer test_model = UTransformer.from_pretrained_backbone( "facebook/dinov3-vits16-pretrain-lvd1689m" ).to("cuda:3") print(test_model)