Add backoff on exceptions from OpenAI API & make pip requirements up-…

…to-date
JetBrains-Research · Jan 16, 2024 · f754cb9 · f754cb9
1 parent 62f54bf
commit f754cb9
Show file tree

Hide file tree

Showing 4 changed files with 905 additions and 834 deletions.
diff --git a/commit_message_generation/poetry.lock b/commit_message_generation/poetry.lock
diff --git a/commit_message_generation/pyproject.toml b/commit_message_generation/pyproject.toml
@@ -28,6 +28,7 @@ bitsandbytes = "^0.41.2"
 accelerate = "^0.23.0"
 optimum = "^1.14.1"
 scipy = "^1.11.4"
+backoff = "^2.2.1"
 
 [tool.poetry.group.dev.dependencies]
 black = {extras = ["jupyter"], version = "^23.7.0"}