KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use transformers as optional dependency (#360)

This commit is contained in:
Albert Villanova del Moral 2025-01-27 10:07:29 +01:00 committed by GitHub
parent ce763ff756
commit 39133acc90
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/smolagents/models.py
View File

@ -24,13 +24,8 @@ from enum import Enum
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
from huggingface_hub import InferenceClient from huggingface_hub import InferenceClient
from huggingface_hub.utils import is_torch_available
from PIL import Image from PIL import Image
from transformers import (
AutoModelForImageTextToText,
AutoProcessor,
StoppingCriteriaList,
is_torch_available,
)
from .tools import Tool from .tools import Tool
from .utils import _is_package_available, encode_image_base64, make_image_url 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]'`" "Please install 'transformers' extra to use 'TransformersModel': `pip install 'smolagents[transformers]'`"
) )
import torch import torch
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoModelForImageTextToText, AutoProcessor, AutoTokenizer
default_model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct" default_model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
if model_id is None: if model_id is None: