Skip to content

Commit e591777

Browse files
Update predict.py
1 parent 80833c4 commit e591777

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dspy/predict/predict.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,9 +77,9 @@ 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

82-
if (temperature is None or temperature <= 0.15) and num_generations is not None and num_generations > 1:
82+
if (temperature is None or temperature <= 0.15) 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)