Fix easy patch in tests

huggingface · Mar 3, 2025 · e40a6a0 · e40a6a0
1 parent 3e6bbf6
commit e40a6a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/smolagents/models.py b/src/smolagents/models.py
@@ -24,7 +24,6 @@
 from enum import Enum
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
-from huggingface_hub import InferenceClient
 from huggingface_hub.utils import is_torch_available
 from PIL import Image
 
@@ -429,6 +428,8 @@ def __init__(
         custom_role_conversions: Optional[Dict[str, str]] = None,
         **kwargs,
     ):
+        from huggingface_hub import InferenceClient
+
         super().__init__(**kwargs)
         self.model_id = model_id
         self.provider = provider