From 39133acc90e2aba6131f761f48b405735e3a873e Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Mon, 27 Jan 2025 10:07:29 +0100
Subject: [PATCH] Use transformers as optional dependency (#360)

---
 src/smolagents/models.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/smolagents/models.py b/src/smolagents/models.py
index eb613df..9a43005 100644
--- a/src/smolagents/models.py
+++ b/src/smolagents/models.py
@@ -24,13 +24,8 @@ 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
-from transformers import (
-    AutoModelForImageTextToText,
-    AutoProcessor,
-    StoppingCriteriaList,
-    is_torch_available,
-)
 
 from .tools import Tool
 from .utils import _is_package_available, encode_image_base64, make_image_url
@@ -463,7 +458,7 @@ class TransformersModel(Model):
                 "Please install 'transformers' extra to use 'TransformersModel': `pip install 'smolagents[transformers]'`"
             )
         import torch
-        from transformers import AutoModelForCausalLM, AutoTokenizer
+        from transformers import AutoModelForCausalLM, AutoModelForImageTextToText, AutoProcessor, AutoTokenizer
 
         default_model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
         if model_id is None: