upgrade flax after installing transformers

cgarciae · cgarciae · commit 2047050e6220 · 2022-10-20T15:19:59.000Z
diff --git a/docs/guides/transfer_learning.ipynb b/docs/guides/transfer_learning.ipynb
@@ -37,7 +37,7 @@
    "outputs": [],
    "source": [
     "! pip install transformers[flax]\n",
-    "! pip install -U jax jaxlib"
+    "! pip install -U flax jax jaxlib"
    ]
   },
   {
diff --git a/docs/guides/transfer_learning.md b/docs/guides/transfer_learning.md
@@ -34,7 +34,7 @@ You will learn how to do transfer learning for simple classification. The steps
 :tags: [skip-execution]
 
 ! pip install transformers[flax]
-! pip install -U jax jaxlib
+! pip install -U flax jax jaxlib
 ```
 
 ## Prepare the model

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@`
`37`	`37`	`"outputs": [],`
`38`	`38`	`"source": [`
`39`	`39`	`"! pip install transformers[flax]\n",`
`40`		`- "! pip install -U jax jaxlib"`
	`40`	`+ "! pip install -U flax jax jaxlib"`
`41`	`41`	`]`
`42`	`42`	`},`
`43`	`43`	`{`