浏览代码

[Bugfix] Fix hugging face command (#944)

Deshraj Yadav 1 年之前
父节点
当前提交
b144c7dccc
共有 3 个文件被更改,包括 3 次插入3 次删除
  1. 1 1
      docs/components/llms.mdx
  2. 1 1
      embedchain/llm/huggingface.py
  3. 1 1
      pyproject.toml

+ 1 - 1
docs/components/llms.mdx

@@ -237,7 +237,7 @@ llm:
 Install related dependencies using the following command:
 
 ```bash
-pip install --upgrade 'embedchain[huggingface_hub]'
+pip install --upgrade 'embedchain[huggingface-hub]'
 ```
 
 First, set `HUGGINGFACE_ACCESS_TOKEN` in environment variable which you can obtain from [their platform](https://huggingface.co/settings/tokens).

+ 1 - 1
embedchain/llm/huggingface.py

@@ -20,7 +20,7 @@ class HuggingFaceLlm(BaseLlm):
         except ModuleNotFoundError:
             raise ModuleNotFoundError(
                 "The required dependencies for HuggingFaceHub are not installed."
-                'Please install with `pip install --upgrade "embedchain[huggingface_hub]"`'
+                'Please install with `pip install --upgrade "embedchain[huggingface-hub]"`'
             ) from None
 
         super().__init__(config=config)

+ 1 - 1
pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "embedchain"
-version = "0.1.7"
+version = "0.1.8"
 description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data"
 authors = [
     "Taranjeet Singh <taranjeet@embedchain.ai>",