Skip to content

Commit 80833c4

Browse files
author
Misha Smirnov
committed
(dspy/predict): corrected temperature check for bedrock claude models
1 parent 4144afa commit 80833c4

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
@@ -79,7 +79,7 @@ def forward(self, **kwargs):
7979
if num_generations is None:
8080
num_generations = lm.kwargs.get("n", lm.kwargs.get("num_generations", None))
8181

82-
if (temperature is None or temperature <= 0.15) and num_generations > 1:
82+
if (temperature is None or temperature <= 0.15) and num_generations is not None and num_generations > 1:
8383
config["temperature"] = 0.7
8484
# print(f"#> Setting temperature to 0.7 since n={num_generations} and prior temperature={temperature}.")
8585

0 commit comments

Comments
 (0)