From bd2418145c029fbe748a4b6b33ac7d9cfc1f6fe7 Mon Sep 17 00:00:00 2001 From: Rishi Puri Date: Thu, 19 Dec 2024 11:43:50 -0800 Subject: [PATCH] Update torch_geometric/nn/nlp/txt2kg.py Co-authored-by: Serge Panev --- torch_geometric/nn/nlp/txt2kg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_geometric/nn/nlp/txt2kg.py b/torch_geometric/nn/nlp/txt2kg.py index 5e668357fc0a..7c5a9bcf8ada 100644 --- a/torch_geometric/nn/nlp/txt2kg.py +++ b/torch_geometric/nn/nlp/txt2kg.py @@ -22,7 +22,7 @@ class TXT2KG(): Use local_lm flag for local debugging/dev. You still need to be able to inference a 14B param LLM, 'VAGOsolutions/SauerkrautLM-v2-14b-DPO'. Smaller LLMs did not work at all in testing. - Note this 14B model requires a considerable amount of VRAM. + Note this 14B model requires a considerable amount of GPU memory. """ def __init__( self,