diff --git a/pyproject.toml b/pyproject.toml index 786751c..728b879 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "refchecker" -version = "0.2.6" +version = "0.2.7" description = "RefChecker provides automatic checking pipeline for detecting fine-grained hallucinations generated by Large Language Models." authors = [ "Xiangkun Hu ", diff --git a/refchecker/extractor/llm_extractor.py b/refchecker/extractor/llm_extractor.py index aad520a..79a26b7 100644 --- a/refchecker/extractor/llm_extractor.py +++ b/refchecker/extractor/llm_extractor.py @@ -102,6 +102,7 @@ def extract_claim_triplets( batch_questions=None, max_new_tokens=500, sagemaker_client=None, + sagemaker_params=None, **kwargs ): """Extract KG triplets from the response text.