Kernels:

aws-neuron
/

vit-encoder-neuron-kernels

jburtoft commited on about 18 hours ago

Commit

986a3a9

1 Parent(s): a95cd4f

examples: use aws-neuron/vit-encoder-neuron-kernels repo id (docs only)

Files changed (1) hide show

examples/dinov3_features.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 from kernels import get_kernel
 from transformers import AutoModel
-k = get_kernel("jburtoft/vit-encoder-neuron-kernels", version=1, trust_remote_code=True)
 model = AutoModel.from_pretrained("facebook/dinov3-vitl16-pretrain-lvd1689m", dtype=torch.float32).eval()
 ref_model = AutoModel.from_pretrained("facebook/dinov3-vitl16-pretrain-lvd1689m", dtype=torch.float32).eval()
 model = model.to("neuron")

 from kernels import get_kernel
 from transformers import AutoModel
+k = get_kernel("aws-neuron/vit-encoder-neuron-kernels", version=1, trust_remote_code=True)
 model = AutoModel.from_pretrained("facebook/dinov3-vitl16-pretrain-lvd1689m", dtype=torch.float32).eval()
 ref_model = AutoModel.from_pretrained("facebook/dinov3-vitl16-pretrain-lvd1689m", dtype=torch.float32).eval()
 model = model.to("neuron")