AutoModelForConditionalGeneration -> T5ForConditionalGeneration
#3
by gojiteji - opened
It seems that AutoModelForConditionalGeneration doesn't exist in HF API, so I changed to T5ForConditionalGeneration based on config.json .
Thanks a mile for the fix!
ybelkada changed pull request status to merged