deley
Browse files- core/generation_utils.py +1 -1
core/generation_utils.py
CHANGED
|
@@ -18,7 +18,7 @@ from tenacity import retry, stop_after_attempt, wait_incrementing
|
|
| 18 |
|
| 19 |
|
| 20 |
|
| 21 |
-
@retry(stop=stop_after_attempt(3), wait=wait_incrementing(increment=
|
| 22 |
def cot_or_da_func(problem: str, llm: BaseLLM = None, **kwargs) -> COTorDAPromptOutput:
|
| 23 |
|
| 24 |
cot_decision_message = [
|
|
|
|
| 18 |
|
| 19 |
|
| 20 |
|
| 21 |
+
@retry(stop=stop_after_attempt(3), wait=wait_incrementing(increment=2))
|
| 22 |
def cot_or_da_func(problem: str, llm: BaseLLM = None, **kwargs) -> COTorDAPromptOutput:
|
| 23 |
|
| 24 |
cot_decision_message = [
|