From 31c8d62ac2e868d6832ff5ddf3d9fb6a026c0d18 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 15 Feb 2024 18:38:03 -0800
Subject: [PATCH] fix(test_async_fn.py): handle hf tgi model not being loaded
 yet

---
 litellm/tests/test_async_fn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/tests/test_async_fn.py b/litellm/tests/test_async_fn.py
index 816ef15093..86cbfafbf1 100644
--- a/litellm/tests/test_async_fn.py
+++ b/litellm/tests/test_async_fn.py
@@ -192,6 +192,8 @@ async def test_hf_completion_tgi():
         )
         # Add any assertions here to check the response
         print(response)
+    except litellm.APIError as e:
+        pass
     except litellm.Timeout as e:
         pass
     except Exception as e: