Getting error when following Flax version
#1
by
aryan1107
- opened
I get this error when following instruction mentioned on model card for Flax version
Thanks for the message! It looks like I made a mistake and had the tokenizer return the tensors as "pt". For flax they need to be "np"
old:inputs = tokenizer(text, return_tensors="pt")
new:inputs = tokenizer(text, return_tensors="np")
I made a change to the model card. Thank you!
nbroad
changed discussion status to
closed