Pin kornia dependency at 0.6.1

nova-land · Dec 14, 2021 · 9990c1b · 9990c1b
1 parent 9937074
commit 9990c1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/text2art.ipynb b/text2art.ipynb
@@ -40,7 +40,7 @@
         "  !rm -Rf clipit\n",
         "  !git clone https://github.com/mfrashad/clipit.git\n",
         "  !pip install ftfy regex tqdm omegaconf pytorch-lightning\n",
-        "  !pip install kornia\n",
+        "  !pip install kornia==0.6.1\n",
         "  !pip install imageio-ffmpeg   \n",
         "  !pip install einops\n",
         "  !pip install torch-optimizer\n",