Update README.md
add missing imports to examples in README.md
This commit is contained in:
parent
780cf7eeb9
commit
a445e556d8
1 changed files with 2 additions and 0 deletions
|
|
@ -142,6 +142,7 @@ rag = LightRAG(
|
||||||
```python
|
```python
|
||||||
from lightrag.llm import hf_model_complete, hf_embedding
|
from lightrag.llm import hf_model_complete, hf_embedding
|
||||||
from transformers import AutoModel, AutoTokenizer
|
from transformers import AutoModel, AutoTokenizer
|
||||||
|
from lightrag.utils import EmbeddingFunc
|
||||||
|
|
||||||
# Initialize LightRAG with Hugging Face model
|
# Initialize LightRAG with Hugging Face model
|
||||||
rag = LightRAG(
|
rag = LightRAG(
|
||||||
|
|
@ -172,6 +173,7 @@ Then you only need to set LightRAG as follows:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
from lightrag.llm import ollama_model_complete, ollama_embedding
|
from lightrag.llm import ollama_model_complete, ollama_embedding
|
||||||
|
from lightrag.utils import EmbeddingFunc
|
||||||
|
|
||||||
# Initialize LightRAG with Ollama model
|
# Initialize LightRAG with Ollama model
|
||||||
rag = LightRAG(
|
rag = LightRAG(
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue