Saif, now the model seems to running normal. I don’t know if still the issue is resolved but it is working fine now. 
but twice while running it ran into same error
Please restart your PC/Laptop so Colab revert back to the original version and then run your notebook without updating anything.