Why would you do an extension for that?  At least for the federal, it is tax exempt income so even if you choose wrong, you aren't creating a tax liability in the wrong year.  Personally, I'm going with the logic that if the loan wasn't forgiven in 2020, you still had a loan at 12/31/20 and the tax exempt income won't hit until 2021.