Skip to content

Commit 39133ac

Browse files
Use transformers as optional dependency (#360)
1 parent ce763ff commit 39133ac

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

src/smolagents/models.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,8 @@
2424
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
2525

2626
from huggingface_hub import InferenceClient
27+
from huggingface_hub.utils import is_torch_available
2728
from PIL import Image
28-
from transformers import (
29-
AutoModelForImageTextToText,
30-
AutoProcessor,
31-
StoppingCriteriaList,
32-
is_torch_available,
33-
)
3429

3530
from .tools import Tool
3631
from .utils import _is_package_available, encode_image_base64, make_image_url
@@ -463,7 +458,7 @@ def __init__(
463458
"Please install 'transformers' extra to use 'TransformersModel': `pip install 'smolagents[transformers]'`"
464459
)
465460
import torch
466-
from transformers import AutoModelForCausalLM, AutoTokenizer
461+
from transformers import AutoModelForCausalLM, AutoModelForImageTextToText, AutoProcessor, AutoTokenizer
467462

468463
default_model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
469464
if model_id is None:

0 commit comments

Comments
 (0)