Small fixes in the Colab

[mrm8488]

Hi, Patrick!
First of all, great colab!
Second, there are a couple of fixes I had to do to get it working:

- config = ReformerConfig(**config
+ config = ReformerConfig(**config) # Add parentheses

- !%%writefile setup.sh #(It is not needed)
+ # After install apex Environment should be restarted

Thanks for your work!

[patrickvonplaten]

Thanks a lot!

[mrm8488]

You are welcome!!