Bug in Colab: Tutorial: Rectified Flow with Neural Network.ipynb

[ernestchu]

Hi, thanks for your great work. I especially appreciate your intuitive blog post. However, I just want to let you know that there's a tiny bug in the attached colab example, Tutorial: Rectified Flow with Neural Network.ipynb.

The selected variable should be diffusion, otherwise it overwrites the rectified_flow_1 in the subsequent blocks.

[ernestchu]

In addition, in draw_plot_initial, I suggest to modify the selected code from 8 to 11 to provide a clearer picture for readers.

8	11