Merge branch 'main' of github.com:danielz02/helm

danielz02 · Oct 20, 2023 · d04e91c · d04e91c
2 parents 5d87a03 + 9aa2166
commit d04e91c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/helm/proxy/clients/huggingface_client.py b/src/helm/proxy/clients/huggingface_client.py
@@ -28,6 +28,8 @@
 )
 from threading import Lock
 
+from ...common.request import ErrorFlags
+
 
 class StopOnToken(StoppingCriteria):
     """
@@ -269,7 +271,10 @@ def do_it():
             response, cached = self.cache.get(cache_key, wrap_request_time(do_it))
         except Exception as e:  # Do something if error is encountered.
             error: str = f"HuggingFace error: {e}"
-            return RequestResult(success=False, cached=False, error=error, completions=[], embedding=[])
+            return RequestResult(
+                success=False, cached=False, error=error, completions=[], embedding=[],
+                error_flags=ErrorFlags(is_retriable=False, is_fatal=False),
+            )
 
         completions = []
         for raw_completion in response["completions"]: