the model facebook/bart-large-cnn is not working properly while loading it using the class TFBartForConditionalGeneration
I tried using the model: facebook/bart-large-cnn by loading it using the class TFBartForConditionalGeneration and TFAutoModelForSeq2SeqLM and return tensor = "tf" in the tokenizer. while using the model in predictions the model is not working properly and not getting the expected results.
pls find the code,
The model is working fine when loaded using the class, model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn") and tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn") and the return tensor ="pt" in tokenizer.
but when used the return tensor as "tf" getting the below error message,
tokenizer([input_text], return_tensors="tf")
kindly help how to use the model in tensorflow.
I have the same issue while using the pipeline API
@Rocketknight1 for me it's working again, no idea why it got messed up temporarily
Hi @levani13 , that's good to hear! We suspect the cause was an issue in loading safetensors in TF. This was fixed by the PR here which was included in in version 4.36. If you updated transformers in the meantime, that might be what happened.
@aron1016 , let me know if upgrading transformers fixes your issues as well!
Yeah I did update it as part of troubleshooting
!pip install transformers==4.36.1
The above code is working fine after upgrading the transformer
thank you