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

Sign up or log in to comment