How to load this model with pytorch?

#3
by thesby - opened

onnxruntime-gpu is too slow. How can I load this model with pytorch ?

Check out under the "Inference code examples" in the model card, the TIMM one.

Thank you. I got it.

model = timm.create_model('hf-hub:SmilingWolf/wd-eva02-large-tagger-v3', pretrained=False).eval()
thesby changed discussion status to closed
thesby changed discussion status to open

Sign up or log in to comment