arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Install the transformers library when running on Colab

Browse Source
This commit is contained in:
Aurélien Geron
2021-03-02 11:10:15 +13:00
parent 0a55f55372
commit 3d418c0308
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
16_nlp_with_rnns_and_attention.ipynb
View File

@@ -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\")"