From 3d418c030809b5f7fb8875d210d9958fbf14e107 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Tue, 2 Mar 2021 11:10:15 +1300 Subject: [PATCH] Install the transformers library when running on Colab --- 16_nlp_with_rnns_and_attention.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/16_nlp_with_rnns_and_attention.ipynb b/16_nlp_with_rnns_and_attention.ipynb index 3dd2d1f..240cd52 100644 --- a/16_nlp_with_rnns_and_attention.ipynb +++ b/16_nlp_with_rnns_and_attention.ipynb @@ -57,6 +57,7 @@ " # %tensorflow_version only exists in Colab.\n", " %tensorflow_version 2.x\n", " !pip install -q -U tensorflow-addons\n", + " !pip install -q -U transformers\n", " IS_COLAB = True\n", "except Exception:\n", " IS_COLAB = False\n", @@ -2615,7 +2616,6 @@ "metadata": {}, "outputs": [], "source": [ - "!pip install -q -U transformers\n", "from transformers import TFOpenAIGPTLMHeadModel\n", "\n", "model = TFOpenAIGPTLMHeadModel.from_pretrained(\"openai-gpt\")"