Just use solve. Nothing special is required. The official version of that example is:
https://docs.sciml.ai/SciMLSensitivity/stable/ode_fitting/optimization_ode/
BTW, I thought I removed these examples from the model zoo years ago (Remove from the model-zoo and point README to tutorials by ChrisRackauckas · Pull Request #237 · FluxML/model-zoo · GitHub). How did you find that old version? Is there some blog post with an old link?