True. I agree with you.
But it might be confusing for beginners (in python) / someone who is just following along the tutorial.
May be
# Pass all parameters required while initializing the model here.
model = TheModelClass()

or we can just
# Modify the arguments passed to the model as needed.
model = TheModelClass(*args, **kwargs)

Let me know.

Also, I just went to the Colab notebook to make sure that everything works in Colab.
But I saw, there are no code cells at all!!!
Here: Colab Notebook 21
Here is the Github notebook: Github Notebook 16
??!