Spaces:

Hazem
/

Fac256xc

Runtime error

boris commited on Apr 20, 2022

Commit

dca3ada

1 Parent(s): 6b84155

style: reformat

Files changed (1) hide show

tools/inference/inference_pipeline.ipynb CHANGED Viewed

@@ -116,9 +116,7 @@
     "from transformers import CLIPProcessor, FlaxCLIPModel\n",
     "\n",
     "# Load dalle-mini\n",
-    "model = DalleBart.from_pretrained(\n",
-    "    DALLE_MODEL, revision=DALLE_COMMIT_ID\n",
-    ")\n",
     "\n",
     "# Load VQGAN\n",
     "vqgan = VQModel.from_pretrained(VQGAN_REPO, revision=VQGAN_COMMIT_ID)\n",

     "from transformers import CLIPProcessor, FlaxCLIPModel\n",
     "\n",
     "# Load dalle-mini\n",
+    "model = DalleBart.from_pretrained(DALLE_MODEL, revision=DALLE_COMMIT_ID)\n",
     "\n",
     "# Load VQGAN\n",
     "vqgan = VQModel.from_pretrained(VQGAN_REPO, revision=VQGAN_COMMIT_ID)\n",