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

gnobitab / RectifiedFlow

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

964 stars 58 forks source link

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

Open ernestchu opened 5 months ago

ernestchu commented 5 months ago

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 commented 5 months ago

In addition, in draw_plot_initial, I suggest to modify the selected code from 8 to 11 to provide a clearer picture for readers. Screenshot 2024-06-17 at 11 46 33 AM

8	11