From 9ac2ed2dc7d8957e77a8b905c36d2ab8cdfbb572 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Mon, 11 Jul 2022 12:30:10 -0400
Subject: [PATCH] add temperature parameter

---
 min_dalle.ipynb | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index c029e63..8b8a41d 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -130,7 +130,6 @@
       "source": [
         "dtype = \"float16\" #@param [\"float32\", \"float16\", \"bfloat16\"]\n",
         "from IPython.display import display, update_display\n",
-        "from math import log2\n",
         "import torch\n",
         "from min_dalle import MinDalle\n",
         "\n",
@@ -204,8 +203,8 @@
         "    seed=-1,\n",
         "    grid_size=grid_size,\n",
         "    log2_mid_count=log2_mid_count,\n",
-        "    log2_k=int(log2(top_k)),\n",
-        "    log2_supercondition_factor=log2(supercondition_factor)\n",
+        "    top_k=int(top_k),\n",
+        "    supercondition_factor=float(supercondition_factor)\n",
         ")\n",
         "\n",
         "is_first = True\n",