Skip to content

Commit c40340d

Browse files
committed
fix parameter errors when invoking bedrock llms
1 parent a5e03d9 commit c40340d

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

pyproject.toml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "refchecker"
3-
version = "0.2.13"
3+
version = "0.2.14"
44
description = "RefChecker provides automatic checking pipeline for detecting fine-grained hallucinations generated by Large Language Models."
55
authors = [
66
"Xiangkun Hu <[email protected]>",
@@ -15,7 +15,7 @@ license = "Apache-2.0"
1515
[tool.poetry.dependencies]
1616
python = "^3.10"
1717
spacy = "^3.7"
18-
boto3 = "^1.34"
18+
boto3 = "^1.35"
1919
torch = "^2"
2020
transformers = "^4.41"
2121
rank-bm25 = "^0.2"
@@ -26,7 +26,7 @@ nltk = "^3.8"
2626
pytorch_lightning = "^2.3" # for alignscore
2727
scikit-learn = "^1.5"
2828
accelerate = "^0.31"
29-
litellm = "^1.48"
29+
litellm = "^1.49"
3030
diskcache = "^5"
3131

3232
# optional dependencies required by specific modules

refchecker/utils.py

+1
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@ def get_model_batch_response(
150150
n=n_choices,
151151
max_tokens=max_new_tokens,
152152
api_base=api_base,
153+
max_workers=None,
153154
**kwargs
154155
)
155156
try:

0 commit comments

Comments
 (0)