This commit is contained in:
August Moharrami 2025-01-31 16:10:16 +03:30 committed by GitHub
parent f1a9b83443
commit b228ffa328
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions

View File

@ -411,9 +411,9 @@ class HfApiModel(Model):
class TransformersModel(Model): class TransformersModel(Model):
"""A class to interact with Hugging Face's Inference API for language model interaction. """A class that uses Hugging Face's Transformers library for language model interaction.
This model allows you to communicate with Hugging Face's models using the Inference API. It can be used in both serverless mode or with a dedicated endpoint, supporting features like stop sequences and grammar customization. This model allows you to load and use Hugging Face's models locally using the Transformers library. It supports features like stop sequences and grammar customization.
> [!TIP] > [!TIP]
> You must have `transformers` and `torch` installed on your machine. Please run `pip install smolagents[transformers]` if it's not the case. > You must have `transformers` and `torch` installed on your machine. Please run `pip install smolagents[transformers]` if it's not the case.