Add Hugging Face Hub LLM support (#762)

This commit is contained in:
Sidharth Mohanty
2023-10-10 00:45:22 +05:30
committed by GitHub
parent e226a89637
commit 0cb78b9067
3 changed files with 117 additions and 0 deletions

View File

@@ -112,6 +112,7 @@ pillow = { version = "10.0.1", optional = true }
torchvision = { version = ">=0.15.1, !=0.15.2", optional = true }
ftfy = { version = "6.1.1", optional = true }
regex = { version = "2023.8.8", optional = true }
huggingface_hub = { version = "^0.17.3", optional = true }
[tool.poetry.group.dev.dependencies]
black = "^23.3.0"
@@ -136,6 +137,7 @@ discord = ["discord"]
slack = ["slack-sdk", "flask"]
whatsapp = ["twilio", "flask"]
images = ["torch", "ftfy", "regex", "pillow", "torchvision"]
huggingface_hub=["huggingface_hub"]
cohere = ["cohere"]
[tool.poetry.group.docs.dependencies]