search

Shivanandroy / simpleT5

simpleT5 is built on top of PyTorch-lightning⚡️ and Transformers🤗 that lets you quickly train your T5 models.
MIT License
386 stars 62 forks source link

How to use flan-t5? #51

Open Bachstelze opened 1 year ago

Bachstelze commented 1 year ago

Is it possible to use the flan models like https://huggingface.co/google/flan-t5-base ?

sr5434 commented 1 year ago

Yes, it is

Bachstelze commented 1 year ago

Okay thanks, I had a typo and missed the "google" in the model path "google/flan-t5-base". It would be lovely to add it to the usage support like: # load (supports t5, google/mt5, google/byt5 and google/flan-t5 models)

sr5434 commented 1 year ago

That sounds like a good idea.