Skip to content

Commit

Permalink
fix parameter errors when invoking bedrock llms
Browse files Browse the repository at this point in the history
  • Loading branch information
HuXiangkun committed Oct 12, 2024
1 parent a5e03d9 commit c40340d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "refchecker"
version = "0.2.13"
version = "0.2.14"
description = "RefChecker provides automatic checking pipeline for detecting fine-grained hallucinations generated by Large Language Models."
authors = [
"Xiangkun Hu <[email protected]>",
Expand All @@ -15,7 +15,7 @@ license = "Apache-2.0"
[tool.poetry.dependencies]
python = "^3.10"
spacy = "^3.7"
boto3 = "^1.34"
boto3 = "^1.35"
torch = "^2"
transformers = "^4.41"
rank-bm25 = "^0.2"
Expand All @@ -26,7 +26,7 @@ nltk = "^3.8"
pytorch_lightning = "^2.3" # for alignscore
scikit-learn = "^1.5"
accelerate = "^0.31"
litellm = "^1.48"
litellm = "^1.49"
diskcache = "^5"

# optional dependencies required by specific modules
Expand Down
1 change: 1 addition & 0 deletions refchecker/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,6 +150,7 @@ def get_model_batch_response(
n=n_choices,
max_tokens=max_new_tokens,
api_base=api_base,
max_workers=None,
**kwargs
)
try:
Expand Down

0 comments on commit c40340d

Please sign in to comment.