From 7e067d2ac549d8e4415fcfa76171fb97d445ac65 Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Sat, 28 Oct 2023 00:35:45 +0200
Subject: [PATCH] Requirements for T5

---
 README.md        | 2 +-
 requirements.txt | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f336e5d..342a5ad 100644
--- a/README.md
+++ b/README.md
@@ -64,7 +64,7 @@ Loaded in bnb4bit mode, it only takes around 6GB VRAM, making it work with 12GB
 
 On windows, you may need a newer version of bitsandbytes for 4bit. Try `python -m pip install bitsandbytes --prefer-binary --extra-index-url=https://jllllll.github.io/bitsandbytes-windows-webui`
 
-You may also need to upgrade transformers. `pip install --upgrade transformers`
+You may also need to upgrade transformers and install spiece for the tokenizer. `pip install -r requirements.txt`
 
 
 ## VAE
diff --git a/requirements.txt b/requirements.txt
index 94c9777..b84828b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,3 @@
-timm==0.6.13
\ No newline at end of file
+timm==0.6.13
+sentencepiece>=0.1.97
+transformers>=4.34.1