Model File Missing Error

#6
by elligottmc - opened

Getting "protectai/deberta-v3-base-prompt-injection-v2 does not appear to have a file named pytorch_model.bin, tf_model.h5, model.ckpt or flax_model.msgpac" error and don't see those files present. Can you share alternate instructions that will work or otherwise advise?

Protect AI org

Hey @elligottmc , thanks for the question. How do you run the model? If you use transformers, it looks for .safetensors format

Thanks @asofter I was able to fix by adding the "safetensors=True" parameter in transformers. Thanks again!

asofter changed discussion status to closed

Sign up or log in to comment