Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 10, 2024
1 parent afc1ecf commit 3f16243
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions torch_geometric/nn/nlp/txt2kg.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,10 @@


class TXT2KG():
"""
Uses NVIDIA NIMs + Prompt engineering to extract KG from text.
"""Uses NVIDIA NIMs + Prompt engineering to extract KG from text.
`nvidia/llama-3.1-nemotron-70b-instruct` is on par or better than GPT4o
in benchmarks. We need a high quality model to ensure high quality KG.
Otherwise we have garbage in garbage out for the rest of the
Otherwise we have garbage in garbage out for the rest of the
GNN+LLM RAG pipeline.
Use local_lm flag for local debugging/dev. You still need to be able to
Expand Down

0 comments on commit 3f16243

Please sign in to comment.