feat: add customization parameters

dev/inference/wandb-backend.ipynb

@@ -32,12 +32,15 @@
    "outputs": [],
    "source": [
     "run_ids = ['3kaut6e8']\n",
+    "# Alamy - 3kaut6e8\n",
+    "# YFCC - to do\n",
+    "# HF spaces - 4oh3u7ca\n",
     "ENTITY, PROJECT = 'wandb', 'hf-flax-dalle-mini'\n",
     "VQGAN_REPO, VQGAN_COMMIT_ID = 'dalle-mini/vqgan_imagenet_f16_16384', None\n",
     "normalize_text = False\n",
     "latest_only = True   # log only latest or all versions\n",
     "suffix = ''           # mainly for duplicate inference runs with a deleted version\n",
-    "add_clip_32 = True"
+    "add_clip_32 = False"
    ]
   },
   {
@@ -59,6 +62,19 @@
     "add_clip_32 = False"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "71f27b96-7e6c-4472-a2e4-e99a8fb67a72",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# model.generate parameters - Not used yet\n",
+    "top_k = None\n",
+    "top_p = None\n",
+    "temperature = None"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -327,6 +343,7 @@
     "                    top_images = [wandb.Image(cur_images[x]) for x in idx]\n",
     "                    top_scores = [scores[x] for x in idx]\n",
     "                    results32.append([sample] + top_images + top_scores)\n",
+    "            pbar.close()\n",
     "\n",
     "    # log results\n",
     "    table = wandb.Table(columns=columns, data=results)\n",