Requirements for T5

2023-10-28 00:35:45 +02:00
parent 7a37319826
commit 7e067d2ac5
2 changed files with 4 additions and 2 deletions
@@ -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
@@ -1 +1,3 @@
-timm==0.6.13
+timm==0.6.13
+sentencepiece>=0.1.97
+transformers>=4.34.1