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