Skip to content

Commit 46df3a0

Browse files
Merge pull request #826 from mikeusru/main
(dspy/predict): corrected temperature check for bedrock claude models
2 parents a501ba4 + e591777 commit 46df3a0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dspy/predict/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def forward(self, **kwargs):
7777

7878
num_generations = config.get("n")
7979
if num_generations is None:
80-
num_generations = lm.kwargs.get("n", lm.kwargs.get("num_generations", None))
80+
num_generations = lm.kwargs.get("n", lm.kwargs.get("num_generations", 1))
8181

8282
if (temperature is None or temperature <= 0.15) and num_generations > 1:
8383
config["temperature"] = 0.7

0 commit comments

Comments
 (0)