boris commited on
Commit
6aa30f5
1 Parent(s): 931b52f

fix(inference): update flax version (temporary fix)

Browse files
dev/inference/inference_pipeline.ipynb CHANGED
@@ -6,7 +6,7 @@
6
  "name": "DALL·E mini - Inference pipeline.ipynb",
7
  "provenance": [],
8
  "collapsed_sections": [],
9
- "authorship_tag": "ABX9TyMUjEt1XMLq+6/GhSnVFsSx",
10
  "include_colab_link": true
11
  },
12
  "kernelspec": {
@@ -826,7 +826,7 @@
826
  "id": "uzjAM2GBYpZX"
827
  },
828
  "source": [
829
- "!pip install -q transformers flax\n",
830
  "!pip install -q git+https://github.com/patil-suraj/vqgan-jax.git # VQGAN model in JAX\n",
831
  "!pip install -q git+https://github.com/borisdayma/dalle-mini.git # Model files"
832
  ],
 
6
  "name": "DALL·E mini - Inference pipeline.ipynb",
7
  "provenance": [],
8
  "collapsed_sections": [],
9
+ "authorship_tag": "ABX9TyOvSyRBA5cMjtEixUxYuduO",
10
  "include_colab_link": true
11
  },
12
  "kernelspec": {
 
826
  "id": "uzjAM2GBYpZX"
827
  },
828
  "source": [
829
+ "!pip install -q transformers flax==0.3.5\n",
830
  "!pip install -q git+https://github.com/patil-suraj/vqgan-jax.git # VQGAN model in JAX\n",
831
  "!pip install -q git+https://github.com/borisdayma/dalle-mini.git # Model files"
832
  ],