diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py
index 88cc71c5a7..cb1eb0152f 100644
--- a/litellm/llms/huggingface_restapi.py
+++ b/litellm/llms/huggingface_restapi.py
@@ -28,7 +28,7 @@ def validate_environment(api_key):
 def completion(
     model: str,
     messages: list,
-    api_base: str,
+    api_base: Optional[str],
     model_response: ModelResponse,
     print_verbose: Callable,
     encoding,