Add sentencepiece dependency for the T5 tokenizer.

This commit is contained in:
Ryan Dick 2024-08-07 14:18:19 +00:00
parent 5dd619e137
commit 3599a4a3e4

View File

@ -47,6 +47,8 @@ dependencies = [
"opencv-python==4.9.0.80",
"pytorch-lightning==2.1.3",
"safetensors==0.4.3",
# sentencepiece is required to load T5TokenizerFast (used by FLUX).
"sentencepiece==0.2.0",
"spandrel==0.3.4",
"timm==0.6.13", # needed to override timm latest in controlnet_aux, see https://github.com/isl-org/ZoeDepth/issues/26
"torch==2.2.2",