Add Colab links

lqiang67 · Dec 27, 2024 · 096f3ae · 096f3ae
1 parent 38ff912
commit 096f3ae
Show file tree

Hide file tree

Showing 6 changed files with 122 additions and 20 deletions.
diff --git a/examples/editing_flux_dev.ipynb b/examples/editing_flux_dev.ipynb
@@ -1,5 +1,22 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/lqiang67/rectified-flow/blob/main/examples/editing_flux_dev.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!git clone https://github.com/lqiang67/rectified-flow.git\n",
+    "%cd rectified-flow/"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,

diff --git a/examples/inference_flux_dev.ipynb b/examples/inference_flux_dev.ipynb
@@ -1,5 +1,22 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/lqiang67/rectified-flow/blob/main/examples/inference_flux_dev.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!git clone https://github.com/lqiang67/rectified-flow.git\n",
+    "%cd rectified-flow/"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

diff --git a/examples/interpolation_conversion.ipynb b/examples/interpolation_conversion.ipynb
@@ -1,5 +1,22 @@
 {
   "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "<a href=\"https://colab.research.google.com/github/lqiang67/rectified-flow/blob/main/examples/interpolation_conversion.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "!git clone https://github.com/lqiang67/rectified-flow.git\n",
+        "%cd rectified-flow/"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": 1,
@@ -295,7 +312,7 @@
         "\t\t\"straight to spherical interp\": straight_to_spherical_interp_list,\n",
         "\t\t\"spherical to straight interp\": spherical_to_straight_interp_list,\n",
         "\t},\n",
-        "\tD1_gt_samples=D1[:5000],\n",
+        "\tD1_gt_samples=D1[:2000],\n",
         "\tnum_trajectories=100,\n",
         "\ttitle=\"Interpolated Trajectories Visualization\",\n",
         ")"
@@ -462,7 +479,7 @@
         "        \"1rf straight\": euler_sampler_straight.trajectories,\n",
         "        \"1rf spherical\": euler_sampler_spherical.trajectories,\n",
         "\t},\n",
-        "    D1_gt_samples=D1[:5000],\n",
+        "    D1_gt_samples=D1[:2000],\n",
         "    num_trajectories=100,\n",
         "    title=\"Euler Sampler Visualization\",\n",
         ")"
@@ -547,7 +564,7 @@
         "        \"straight rf\": euler_sampler_straight.trajectories, \n",
         "        \"straight to spherical rf\": euler_sampler_converted_spherical.trajectories\n",
         "    },\n",
-        "\tD1_gt_samples=D1[:5000],\n",
+        "\tD1_gt_samples=D1[:2000],\n",
         "\tnum_trajectories=100,\n",
         "\ttitle=f\"Straight Converted to Spherical   v.s.   Original Straight RF\",\n",
         ")"
@@ -588,7 +605,7 @@
         "        \"spherical rf\": euler_sampler_spherical.trajectories, \n",
         "        \"converted spherical rf\": euler_sampler_converted_spherical.trajectories\n",
         "    },\n",
-        "\tD1_gt_samples=D1[:5000],\n",
+        "\tD1_gt_samples=D1[:2000],\n",
         "\tnum_trajectories=100,\n",
         "\ttitle=f\"Converte Straight to Spherical RF v.s. Spherical RF\",\n",
         ")"
@@ -674,7 +691,7 @@
         "        \"reparam spherical rf\": euler_sampler_spherical.trajectories, \n",
         "        \"straight to spherical rf\": euler_sampler_converted_spherical.trajectories\n",
         "    },\n",
-        "\tD1_gt_samples=D1[:5000],\n",
+        "\tD1_gt_samples=D1[:2000],\n",
         "\tnum_trajectories=100,\n",
         "\ttitle=f\"Re-parametrized Straight RF  v.s.  Converted Straight to Spherical RF\",\n",
         ")"

diff --git a/examples/natural_euler_sampler.ipynb b/examples/natural_euler_sampler.ipynb
@@ -1,5 +1,22 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/lqiang67/rectified-flow/blob/main/examples/natural_euler_sampler.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!git clone https://github.com/lqiang67/rectified-flow.git\n",
+    "%cd rectified-flow/"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
@@ -124,7 +141,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Both vanilla Euler sampler, huge difference in the trajectories\n",
+    "# Both vanilla Euler sampler, noticeable difference in final generated samples\n",
     "\n",
     "from rectified_flow.samplers import EulerSampler\n",
     "\n",
@@ -154,7 +171,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Unmatched time grid, both natural euler samplers, nearly identical\n",
+    "# Unmatched time grid, both natural euler samplers, nearly identical final generated samples\n",
     "\n",
     "from rectified_flow.samplers import CurvedEulerSampler\n",
     "\n",
@@ -184,7 +201,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Matched time grid, both natural euler samplers, exactly the same\n",
+    "# Matched time grid, both natural euler samplers, exactly the same final generated samples\n",
     "\n",
     "def convert_time(t, affine_interp):\n",
     "    return [affine_interp.alpha(t) / (affine_interp.alpha(t) + affine_interp.beta(t)) for t in t]\n",

diff --git a/examples/samplers_2d_toys.ipynb b/examples/samplers_2d_toys.ipynb
@@ -1,5 +1,22 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/lqiang67/rectified-flow/blob/main/examples/samplers_2d_toys.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!git clone https://github.com/lqiang67/rectified-flow.git\n",
+    "%cd rectified-flow/"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -380,7 +397,7 @@
     "\\text{Effective Noise at time } t = \\text{step\\_size} \\times \\text{noise\\_scale} \\times \\beta_t^{\\text{noise\\_decay\\_rate}}\n",
     "$$\n",
     "\n",
-    "Check out this [blog post](https://rectifiedflow.github.io/blog/2024/samplers/) for more details."
+    "Check out this [blog post](https://rectifiedflow.github.io/blog/2024/diffusion/) for more details."
    ]
   },
   {
@@ -463,7 +480,7 @@
     "sde_sampler = OverShootingSampler(\n",
     "    rectified_flow=straight_rf,\n",
     "    num_steps=10,\n",
-    "    num_samples=500,\n",
+    "    num_samples=1000,\n",
     "    c=15.0,\n",
     "    overshooting_method=\"t+dt\"\n",
     ")\n",
@@ -475,7 +492,7 @@
     "# Plot CurvedEulerSampler results\n",
     "visualize_2d_trajectories_plotly(\n",
     "    {\"overshooting\": sde_sampler.trajectories},\n",
-    "    D1[:5000], # D1 defined previously\n",
+    "    D1[:1000], # D1 defined previously\n",
     "    num_trajectories=100,\n",
     "    show_legend=True\n",
     ")\n",

diff --git a/examples/train_2d_toys.ipynb b/examples/train_2d_toys.ipynb
@@ -1,5 +1,22 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/lqiang67/rectified-flow/blob/main/examples/train_2d_toys.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!git clone https://github.com/lqiang67/rectified-flow.git\n",
+    "%cd rectified-flow/"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -64,7 +81,7 @@
     "D1, labels = pi_1.sample_with_labels([n_samples])\n",
     "labels.tolist()\n",
     "\n",
-    "plt.figure(figsize=(5, 5))\n",
+    "plt.figure(figsize=(3, 3))\n",
     "plt.title(r'Samples from $\\pi_0$ and $\\pi_1$')\n",
     "plt.scatter(D0[:, 0].cpu(), D0[:, 1].cpu(), alpha=0.5, label=r'$\\pi_0$')\n",
     "plt.scatter(D1[:, 0].cpu(), D1[:, 1].cpu(), alpha=0.5, label=r'$\\pi_1$')\n",
@@ -132,8 +149,8 @@
     "    \n",
     "visualize_2d_trajectories_plotly(\n",
     "    trajectories_dict={\n",
-    "        \"upper interpolation\": interp_upper,\n",
-    "\t\t\"lower interpolation\": interp_lower\n",
+    "        \"upper\": interp_upper,\n",
+    "\t\t\"lower\": interp_lower\n",
     "    },\n",
     "    D1_gt_samples=torch.cat([x_1_upper, x_1_lower], dim=0),\n",
     "    num_trajectories=100,\n",
@@ -167,7 +184,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -221,7 +238,7 @@
    },
    "outputs": [],
    "source": [
-    "optimizer = torch.optim.Adam(model.parameters(), lr=1e-2)\n",
+    "optimizer = torch.optim.Adam(model.parameters(), lr=1e-3)\n",
     "batch_size = 1024\n",
     "\n",
     "losses = []\n",
@@ -349,7 +366,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -392,7 +409,7 @@
     "\toptimizer.step()\n",
     "\tlosses.append(loss.item())\n",
     "\n",
-    "\tif step % 200 == 0:\n",
+    "\tif step % 1000 == 0:\n",
     "\t\tprint(f\"Epoch {step}, Loss: {loss.item()}\")\n",
     "    \n",
     "plt.plot(losses)"
@@ -449,7 +466,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -483,7 +500,7 @@
     "\toptimizer.step()\n",
     "\tlosses.append(loss.item())\n",
     "\n",
-    "\tif step % 200 == 0:\n",
+    "\tif step % 1000 == 0:\n",
     "\t\tprint(f\"Epoch {step}, Loss: {loss.item()}\")\n",
     "    \n",
     "plt.plot(losses)"