diff --git a/dspy/predict/predict.py b/dspy/predict/predict.py index 17ae56721a..3325600947 100644 --- a/dspy/predict/predict.py +++ b/dspy/predict/predict.py @@ -77,7 +77,7 @@ def forward(self, **kwargs): num_generations = config.get("n") if num_generations is None: - num_generations = lm.kwargs.get("n", lm.kwargs.get("num_generations", None)) + num_generations = lm.kwargs.get("n", lm.kwargs.get("num_generations", 1)) if (temperature is None or temperature <= 0.15) and num_generations > 1: config["temperature"] = 0.7