diff --git a/garak/generators/ollama.py b/garak/generators/ollama.py index b6b8ee533..695203887 100644 --- a/garak/generators/ollama.py +++ b/garak/generators/ollama.py @@ -60,7 +60,7 @@ class OllamaGeneratorChat(OllamaGenerator): @backoff.on_exception( backoff.fibo, - (TimeoutError, ollama.ResponseError), + (TimeoutException, ollama.ResponseError), max_value=70, giveup=_give_up, )