From 569ae28e9e78bd6102d7523964ca7facf94300f2 Mon Sep 17 00:00:00 2001 From: SillyLossy Date: Wed, 1 Mar 2023 21:09:26 +0200 Subject: [PATCH] Add tensorflow to colab --- colab/GPU.ipynb | 1 + 1 file changed, 1 insertion(+) diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 635b89248..900060f55 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -173,6 +173,7 @@ "!git clone https://github.com/SillyLossy/TavernAI-extras\n", "%cd TavernAI-extras\n", "!pip install -r requirements.txt\n", + "!pip install -U tensorflow==2.9.2\n", "extras_process = subprocess.Popen(['python', '/TavernAI-extras/server.py', '--cpu', ShareParam, ModulesParam], stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd='/TavernAI-extras')\n", "print('processId:', extras_process.pid)\n", "\n",