Getting unexpected NotImplementedError when running model

I think I may have found a solution here. I’ll do some more research based off of this answer.