Current code applicable to FlanT5 models as well?

Hi, I was planning to reproduce your work, but I found that the current version of code is mostly targeting LLAMA models.

I'm quite unsure whether the code at the moment also supports experiments based on FlanT5 models (like the paper reports). Is the code applicable to FlanT5 models as well? Thank you for your amazing work, and thanks again in advance for your reply!