Transformers documentation

Auto-классы

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v5.22.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Auto-классы

Во многих случаях архитектуру можно определить по имени или пути предобученной модели, переданной в метод from_pretrained(). Создание экземпляров AutoConfig, AutoModel или AutoTokenizer автоматически порождает соответствующий класс архитектуры. Например:

model = AutoModel.from_pretrained("google-bert/bert-base-cased", device_map="auto")

Это создаёт экземпляр модели BertModel.

Для каждой задачи существует соответствующий класс AutoModel.

Расширение классов Auto

Каждый auto-класс предоставляет метод, позволяющий расширить его с помощью пользовательских классов. Например, если вы определяете пользовательский класс модели NewModel, убедитесь, что вы также определили соответствующий NewModelConfig. Затем вы можете зарегистрировать их в auto-классах, как показано ниже:

from transformers import AutoConfig, AutoModel


AutoConfig.register("new-model", NewModelConfig)
AutoModel.register(NewModelConfig, NewModel)

Затем вы можете использовать классы auto в обычном режиме.

Если ваш NewModelConfig является подклассом PreTrainedConfig, убедитесь, что его атрибут model_type установлен в то же значение, которое вы используете при регистрации конфигурации (здесь "new-model").

Аналогично, если ваш NewModel является подклассом PreTrainedModel, убедитесь, что его атрибут config_class установлен в тот же класс, который вы используете при регистрации модели (здесь NewModelConfig).

AutoConfig

класс transformers.AutoConfig

< >

( )

Это универсальный класс конфигурации, который будет создан как один из классов конфигурации библиотеки при использовании метода класса from_pretrained().

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_pretrained

< >

( pretrained_model_name_or_path: str | os.PathLike[str]**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — может быть одним из следующих:
    • Строка, model id конфигурации предобученной модели, размещённой в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему файл конфигурации, сохранённый с помощью метода save_pretrained() или метода save_pretrained(), например, ./my_model_directory/.
    • путь к сохранённому конфигурационному JSON-файлу, например, ./my_model_directory/configuration.json.
  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Указывает, следует ли принудительно (пере)загружать веса модели и файлы конфигурации, заменяя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • return_unused_kwargs (bool, необязательный, по умолчанию False) — если False, эта функция возвращает только итоговый объект конфигурации.

    Если True, то эта функция возвращает Tuple(config, unused_kwargs), где unused_kwargs — это словарь, состоящий из пар ключ/значение, ключи которых не являются атрибутами конфигурации, то есть та часть kwargs, которая не была использована для обновления config и в остальном игнорируется.

  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • kwargs(additional keyword arguments, optional) — Значения в kwargs для любых ключей, являющихся атрибутами конфигурации, будут использоваться для переопределения загруженных значений. Поведение в отношении пар ключ/значение, ключи которых не являются атрибутами конфигурации, контролируется параметром return_unused_kwargs.

Создайте экземпляр одного из классов конфигурации библиотеки на основе конфигурации предобученной модели.

Класс конфигурации, который будет создан, выбирается на основе свойства model_type объекта конфигурации, который загружается, а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

Примеры:

>>> from transformers import AutoConfig

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-uncased")

>>> # Download configuration from huggingface.co (user-uploaded) and cache.
>>> config = AutoConfig.from_pretrained("dbmdz/bert-base-german-cased")

>>> # If configuration file is in a directory (e.g., was saved using *save_pretrained('./test/saved_model/')*).
>>> config = AutoConfig.from_pretrained("./test/bert_saved_model/")

>>> # Load a specific configuration file.
>>> config = AutoConfig.from_pretrained("./test/bert_saved_model/my_configuration.json")

>>> # Change some config attributes when loading a pretrained config.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-uncased", output_attentions=True, foo=False)
>>> config.output_attentions
True

>>> config, unused_kwargs = AutoConfig.from_pretrained(
...     "google-bert/bert-base-uncased", output_attentions=True, foo=False, return_unused_kwargs=True
... )
>>> config.output_attentions
True

>>> unused_kwargs
{'foo': False}

register

< >

( model_typeconfigexist_ok = False )

Параметры

  • model_type (str) — Тип модели, например «bert» или «gpt».
  • config (PreTrainedConfig) — Конфигурация для регистрации.

Зарегистрируйте новую конфигурацию для этого класса.

AutoTokenizer

класс transformers.AutoTokenizer

< >

( )

Это универсальный класс токенизатора, который при создании с помощью метода класса AutoTokenizer.from_pretrained() будет инстанцирован как один из классов токенизаторов библиотеки.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_pretrained

< >

( pretrained_model_name_or_path*inputs**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть одним из следующих:
    • Строка, model id предопределенного токенизатора, размещенного в репозитории модели на huggingface.co.
    • Путь к каталогу, содержащему файлы словаря, необходимые для токенизатора, например, сохранённые с помощью метода save_pretrained(), например ./my_model_directory/.
    • путь к одному сохранённому файлу словаря, если и только если токенизатору требуется единственный файл словаря (например, как у Bert или XLNet), например: ./my_model_directory/vocab.txt. (Не применимо ко всем производным классам)
  • inputs (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() объекта Tokenizer.
  • config (PreTrainedConfig, необязательный) — Объект конфигурации, используемый для определения класса токенизатора, который будет создан.
  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Указывает, следует ли принудительно (пере)загружать веса модели и файлы конфигурации, заменяя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • subfolder (str, необязательный) — Если соответствующие файлы находятся в подпапке репозитория модели на huggingface.co (например, для facebook/rag-token-base), укажите её здесь.
  • tokenizer_type (str, необязательный) — Тип токенизатора, который будет загружен.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • kwargs (дополнительные именованные аргументы, необязательный) — будут переданы в метод __init__() токенизатора. Можно использовать для установки специальных токенов, таких как bos_token, eos_token, unk_token, sep_token, pad_token, cls_token, mask_token, additional_special_tokens. Подробности о параметрах см. в __init__().

Создайте экземпляр одного из классов токенизаторов библиотеки на основе словаря предобученной модели.

Класс токенизатора, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону в pretrained_model_name_or_path:

Примеры:

>>> from transformers import AutoTokenizer

>>> # Download vocabulary from huggingface.co and cache.
>>> tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")

>>> # Download vocabulary from huggingface.co (user-uploaded) and cache.
>>> tokenizer = AutoTokenizer.from_pretrained("dbmdz/bert-base-german-cased")

>>> # If vocabulary files are in a directory (e.g. tokenizer was saved using *save_pretrained('./test/saved_model/')*)
>>> # tokenizer = AutoTokenizer.from_pretrained("./test/bert_saved_model/")

>>> # Download vocabulary from huggingface.co and define model-specific arguments
>>> tokenizer = AutoTokenizer.from_pretrained("FacebookAI/roberta-base", add_prefix_space=True)

register

< >

( config_classtokenizer_class = Noneslow_tokenizer_class = Nonefast_tokenizer_class = Noneexist_ok = False )

Параметры

  • config_class (PreTrainedConfig) — Конфигурация, соответствующая модели, которую нужно зарегистрировать.
  • tokenizer_class — класс токенизатора для регистрации (V5 — предпочтительный параметр).
  • slow_tokenizer_class — (Устарело) Медленный токенизатор для регистрации.
  • fast_tokenizer_class — (Устарело) Быстрый токенизатор для регистрации.

Зарегистрируйте новый токенизатор в этом сопоставлении.

AutoFeatureExtractor

класс transformers.AutoFeatureExtractor

< >

( )

Это универсальный класс экстрактора признаков, который при создании с помощью метода класса AutoFeatureExtractor.from_pretrained() будет инстанцирован как один из классов экстракторов признаков библиотеки.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_pretrained

< >

( pretrained_model_name_or_path**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Это может быть:
    • строка, model id предобученного feature_extractor, размещенного в репозитории модели на huggingface.co.
    • путь к каталогу, содержащему файл экстрактора признаков, сохранённый с помощью метода save_pretrained(), например, ./my_model_directory/.
    • путь к сохранённому JSON-файлу экстрактора признаков, например, ./my_model_directory/preprocessor_config.json.
  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженный предобученный экстрактор признаков модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать файлы экстрактора признаков и заменять кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • token (str или bool, необязательный) — Токен, используемый в качестве HTTP bearer-авторизации для удалённых файлов. Если True, будет использован токен, сгенерированный при выполнении hf auth login (хранится в ~/.huggingface).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • return_unused_kwargs (bool, необязательный, по умолчанию False) — Если False, функция возвращает только итоговый объект feature extractor. Если True, функция возвращает Tuple(feature_extractor, unused_kwargs), где unused_kwargs — это словарь, содержащий пары ключ/значение, ключи которых не являются атрибутами feature extractor, то есть та часть kwargs, которая не была использована для обновления feature_extractor и в остальном игнорируется.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • kwargs (dict[str, Any], необязательный) — Значения в kwargs для любых ключей, являющихся атрибутами feature extractor, будут использоваться для переопределения загруженных значений. Поведение в отношении пар ключ/значение, ключи которых не являются атрибутами feature extractor, контролируется параметром return_unused_kwargs.

Создайте экземпляр одного из классов feature extractor библиотеки на основе словаря предобученной модели.

Класс экстрактора признаков, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону в pretrained_model_name_or_path:

Передача token=True обязательна, если вы хотите использовать частную модель.

Примеры:

>>> from transformers import AutoFeatureExtractor

>>> # Download feature extractor from huggingface.co and cache.
>>> feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/wav2vec2-base-960h")

>>> # If feature extractor files are in a directory (e.g. feature extractor was saved using *save_pretrained('./test/saved_model/')*)
>>> # feature_extractor = AutoFeatureExtractor.from_pretrained("./test/saved_model/")

register

< >

( config_classfeature_extractor_classexist_ok = False )

Параметры

  • config_class (PreTrainedConfig) — Конфигурация, соответствующая модели, которую нужно зарегистрировать.
  • feature_extractor_class (FeatureExtractorMixin) — Извлекающий признаки, который нужно зарегистрировать.

Зарегистрируйте новый экстрактор признаков для этого класса.

AutoImageProcessor

класс transformers.AutoImageProcessor

< >

( )

Это универсальный класс процессора изображений, который при создании с помощью метода класса AutoImageProcessor.from_pretrained() будет инстанцирован как один из классов процессоров изображений библиотеки.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_pretrained

< >

( pretrained_model_name_or_path*inputs**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Это может быть:
    • строка, model id предобученного image_processor, размещенного в репозитории модели на huggingface.co.
    • путь к каталогу, содержащему файл процессора изображений, сохранённый с помощью метода save_pretrained(), например, ./my_model_directory/.
    • путь к сохранённому JSON-файлу процессора изображений, например, ./my_model_directory/preprocessor_config.json.
  • cache_dir (str or os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженный предобученный процессор изображений модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать файлы процессора изображений и заменять кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • token (str или bool, необязательный) — Токен, используемый в качестве HTTP bearer-авторизации для удалённых файлов. Если True, будет использован токен, сгенерированный при выполнении hf auth login (хранится в ~/.huggingface).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • use_fast (bool, необязательный, по умолчанию False) — Устарело: используйте backend="torchvision". Этот параметр сохранён для обратной совместимости. Использовать быстрый процессор изображений на основе torchvision, если он поддерживается для данной модели. Если для данной модели недоступен быстрый процессор изображений, вместо него возвращается обычный процессор на основе numpy.
  • backend (str, необязательный, по умолчанию None) — Бэкенд для обработки изображений. Допустимые значения:
    • None: автоматически выбрать лучший доступный бэкенд (torchvision, если он доступен, в противном случае pil)
    • "torchvision": Использовать бэкенд Torchvision (ускорение на GPU, быстрее)
    • "pil": Использовать бэкенд PIL (переносимый, только CPU)
    • Любое имя пользовательского бэкенда, зарегистрированное с помощью метода register()
  • return_unused_kwargs (bool, необязательный, по умолчанию False) — Если False, функция возвращает только итоговый объект image processor. Если True, функция возвращает Tuple(image_processor, unused_kwargs), где unused_kwargs — это словарь, содержащий пары ключ/значение, ключи которых не являются атрибутами image processor, то есть та часть kwargs, которая не была использована для обновления image_processor и в остальном игнорируется.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • image_processor_filename (str, необязательный, по умолчанию "config.json") — Имя файла в каталоге модели, используемого для конфигурации процессора изображений.
  • kwargs (dict[str, Any], необязательный) — Значения в kwargs для любых ключей, являющихся атрибутами image processor, используются для переопределения загруженных значений. Поведение в отношении пар ключ/значение, ключи которых не являются атрибутами image processor, контролируется параметром return_unused_kwargs.

Создайте экземпляр одного из классов процессора изображений библиотеки на основе словаря предобученной модели.

Класс процессора изображений, который будет создан, выбирается на основе свойства model_type объекта конфигурации (переданного в качестве аргумента или загруженного из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

  • aimv2 — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель Aimv2Config)
  • aimv2_vision_model — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель Aimv2VisionConfig)
  • align — {'torchvision': 'EfficientNetImageProcessor', 'pil': 'EfficientNetImageProcessorPil'} (модель AlignConfig)
  • altclip — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель AltCLIPConfig)
  • aria — {'pil': 'AriaImageProcessorPil', 'torchvision': 'AriaImageProcessor'} (модель AriaConfig)
  • aya_vision — {'torchvision': 'GotOcr2ImageProcessor', 'pil': 'GotOcr2ImageProcessorPil'} (модель AyaVisionConfig)
  • beit — {'pil': 'BeitImageProcessorPil', 'torchvision': 'BeitImageProcessor'} (модель BeitConfig)
  • bit — {'pil': 'BitImageProcessorPil', 'torchvision': 'BitImageProcessor'} (модель BitConfig)
  • blip — {'pil': 'BlipImageProcessorPil', 'torchvision': 'BlipImageProcessor'} (модель BlipConfig)
  • blip-2 — {'torchvision': 'BlipImageProcessor', 'pil': 'BlipImageProcessorPil'} (модель Blip2Config)
  • bridgetower — {'pil': 'BridgeTowerImageProcessorPil', 'torchvision': 'BridgeTowerImageProcessor'} (модель BridgeTowerConfig)
  • chameleon — {'pil': 'ChameleonImageProcessorPil', 'torchvision': 'ChameleonImageProcessor'} (модель ChameleonConfig)
  • chinese_clip — {'pil': 'ChineseCLIPImageProcessorPil', 'torchvision': 'ChineseCLIPImageProcessor'} (модель ChineseCLIPConfig)
  • chmv2 — {'torchvision': 'CHMv2ImageProcessor'} (модель CHMv2Config)
  • clip — {'pil': 'CLIPImageProcessorPil', 'torchvision': 'CLIPImageProcessor'} (модель CLIPConfig)
  • clipseg — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель CLIPSegConfig)
  • cohere2_vision — {'torchvision': 'Cohere2VisionImageProcessor'} (модель Cohere2VisionConfig)
  • cohere_compass — {'pil': 'CohereCompassImageProcessorPil', 'torchvision': 'CohereCompassImageProcessor'} (модель CohereCompassConfig)
  • colpali — {'torchvision': 'SiglipImageProcessor', 'pil': 'SiglipImageProcessorPil'} (модель ColPaliConfig)
  • colqwen2 — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель ColQwen2Config)
  • conditional_detr — {'pil': 'ConditionalDetrImageProcessorPil', 'torchvision': 'ConditionalDetrImageProcessor'} (модель ConditionalDetrConfig)
  • convnext — {'pil': 'ConvNextImageProcessorPil', 'torchvision': 'ConvNextImageProcessor'} (модель ConvNextConfig)
  • convnextv2 — {'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'} (модель ConvNextV2Config)
  • cosmos3_edge — {'pil': 'Cosmos3EdgeImageProcessorPil', 'torchvision': 'Cosmos3EdgeImageProcessor'} (модель Cosmos3EdgeConfig)
  • cosmos3_omni — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Cosmos3OmniConfig)
  • cvt — {'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'} (модель CvtConfig)
  • data2vec-vision — {'torchvision': 'BeitImageProcessor', 'pil': 'BeitImageProcessorPil'} (модель Data2VecVisionConfig)
  • deepseek_ocr2 — {'pil': 'DeepseekOcr2ImageProcessorPil', 'torchvision': 'DeepseekOcr2ImageProcessor'} (модель DeepseekOcr2Config)
  • deepseek_vl — {'pil': 'DeepseekVLImageProcessorPil', 'torchvision': 'DeepseekVLImageProcessor'} (модель DeepseekVLConfig)
  • deepseek_vl_hybrid — {'pil': 'DeepseekVLHybridImageProcessorPil', 'torchvision': 'DeepseekVLHybridImageProcessor'} (модель DeepseekVLHybridConfig)
  • deformable_detr — {'pil': 'DeformableDetrImageProcessorPil', 'torchvision': 'DeformableDetrImageProcessor'} (модель DeformableDetrConfig)
  • deimv2 — {'torchvision': 'RTDetrImageProcessor', 'pil': 'RTDetrImageProcessorPil'} (модель Deimv2Config)
  • deit — {'pil': 'DeiTImageProcessorPil', 'torchvision': 'DeiTImageProcessor'} (модель DeiTConfig)
  • depth_anything — {'torchvision': 'DPTImageProcessor', 'pil': 'DPTImageProcessorPil'} (модель DepthAnythingConfig)
  • depth_pro — {'torchvision': 'DepthProImageProcessor'} (модель DepthProConfig)
  • detr — {'pil': 'DetrImageProcessorPil', 'torchvision': 'DetrImageProcessor'} (модель DetrConfig)
  • dinat — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель DinatConfig)
  • dinov2 — {'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'} (модель Dinov2Config)
  • dinov3_vit — {'torchvision': 'DINOv3ViTImageProcessor'} (модель DINOv3ViTConfig)
  • donut-swin — {'torchvision': 'DonutImageProcessor', 'pil': 'DonutImageProcessorPil'} (модель DonutSwinConfig)
  • dpt — {'pil': 'DPTImageProcessorPil', 'torchvision': 'DPTImageProcessor'} (модель DPTConfig)
  • edgetam — {'torchvision': 'Sam2ImageProcessor'} (модель EdgeTamConfig)
  • efficientloftr — {'pil': 'EfficientLoFTRImageProcessorPil', 'torchvision': 'EfficientLoFTRImageProcessor'} (модель EfficientLoFTRConfig)
  • efficientnet — {'pil': 'EfficientNetImageProcessorPil', 'torchvision': 'EfficientNetImageProcessor'} (модель EfficientNetConfig)
  • embedding_gemma2 — {'pil': 'Gemma4ImageProcessorPil', 'torchvision': 'Gemma4ImageProcessor'} (модель EmbeddingGemma2Config)
  • emu3 — {'pil': 'Emu3ImageProcessor'} (модель Emu3Config)
  • eomt — {'pil': 'EomtImageProcessorPil', 'torchvision': 'EomtImageProcessor'} (модель EomtConfig)
  • eomt_dinov3 — {'torchvision': 'EomtImageProcessor', 'pil': 'EomtImageProcessorPil'} (модель EomtDinov3Config)
  • ernie4_5_vl_moe — {'pil': 'Ernie4_5_VLMoeImageProcessorPil', 'torchvision': 'Ernie4_5_VLMoeImageProcessor'} (модель Ernie4_5_VLMoeConfig)
  • exaone4_5 — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Exaone4_5_Config)
  • flava — {'pil': 'FlavaImageProcessorPil', 'torchvision': 'FlavaImageProcessor'} (модель FlavaConfig)
  • florence2 — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель Florence2Config)
  • focalnet — {'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'} (модель FocalNetConfig)
  • fuyu — {'pil': 'FuyuImageProcessorPil', 'torchvision': 'FuyuImageProcessor'} (модель FuyuConfig)
  • gemma3 — {'pil': 'Gemma3ImageProcessorPil', 'torchvision': 'Gemma3ImageProcessor'} (модель Gemma3Config)
  • gemma3n — {'torchvision': 'SiglipImageProcessor', 'pil': 'SiglipImageProcessorPil'} (модель Gemma3nConfig)
  • gemma4 — {'pil': 'Gemma4ImageProcessorPil', 'torchvision': 'Gemma4ImageProcessor'} (модель Gemma4Config)
  • gemma4_unified — {'torchvision': 'Gemma4UnifiedImageProcessor'} (модель Gemma4UnifiedConfig)
  • git — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель GitConfig)
  • glm46v — {'pil': 'Glm46VImageProcessorPil', 'torchvision': 'Glm46VImageProcessor'} (модель Glm46VConfig)
  • glm4v — {'pil': 'Glm4vImageProcessorPil', 'torchvision': 'Glm4vImageProcessor'} (модель Glm4vConfig)
  • glm5_next — {'pil': 'Glm5NextImageProcessorPil', 'torchvision': 'Glm5NextImageProcessor'} (модель Glm5NextConfig)
  • glm_image — {'pil': 'GlmImageImageProcessorPil', 'torchvision': 'GlmImageImageProcessor'} (модель GlmImageConfig)
  • glmga — {'pil': 'GlmgaImageProcessorPil', 'torchvision': 'GlmgaImageProcessor'} (модель GlmgaConfig)
  • glpn — {'pil': 'GLPNImageProcessorPil', 'torchvision': 'GLPNImageProcessor'} (модель GLPNConfig)
  • got_ocr2 — {'pil': 'GotOcr2ImageProcessorPil', 'torchvision': 'GotOcr2ImageProcessor'} (модель GotOcr2Config)
  • granite4_vision — {'torchvision': 'LlavaNextImageProcessor', 'pil': 'LlavaNextImageProcessorPil'} (модель Granite4VisionConfig)
  • grounding-dino — {'pil': 'GroundingDinoImageProcessorPil', 'torchvision': 'GroundingDinoImageProcessor'} (модель GroundingDinoConfig)
  • groupvit — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель GroupViTConfig)
  • hiera — {'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'} (модель HieraConfig)
  • hunyuan_vl — {'pil': 'HunYuanVLImageProcessorPil', 'torchvision': 'HunYuanVLImageProcessor'} (модель HunYuanVLConfig)
  • hyperclovax_vision_v2 — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель HyperCLOVAXVisionV2Config)
  • idefics — {'pil': 'IdeficsImageProcessorPil', 'torchvision': 'IdeficsImageProcessor'} (модель IdeficsConfig)
  • idefics2 — {'pil': 'Idefics2ImageProcessorPil', 'torchvision': 'Idefics2ImageProcessor'} (модель Idefics2Config)
  • idefics3 — {'pil': 'Idefics3ImageProcessorPil', 'torchvision': 'Idefics3ImageProcessor'} (модель Idefics3Config)
  • ijepa — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель IJepaConfig)
  • imagegpt — {'pil': 'ImageGPTImageProcessorPil', 'torchvision': 'ImageGPTImageProcessor'} (модель ImageGPTConfig)
  • inkling_mm_model — {'torchvision': 'InklingImageProcessor'} (модель InklingConfig)
  • instructblip — {'torchvision': 'BlipImageProcessor', 'pil': 'BlipImageProcessorPil'} (модель InstructBlipConfig)
  • internvl — {'torchvision': 'GotOcr2ImageProcessor', 'pil': 'GotOcr2ImageProcessorPil'} (модель InternVLConfig)
  • janus — {'pil': 'JanusImageProcessorPil', 'torchvision': 'JanusImageProcessor'} (модель JanusConfig)
  • kimi_k25 — {'torchvision': 'Kimi_K25ImageProcessor'} (модель Kimi_K25Config)
  • kosmos-2 — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель Kosmos2Config)
  • kosmos-2.5 — {'torchvision': 'Kosmos2_5ImageProcessor', 'pil': 'Kosmos2_5ImageProcessorPil'} (модель Kosmos2_5Config)
  • layoutlmv2 — {'pil': 'LayoutLMv2ImageProcessorPil', 'torchvision': 'LayoutLMv2ImageProcessor'} (модель LayoutLMv2Config)
  • layoutlmv3 — {'pil': 'LayoutLMv3ImageProcessorPil', 'torchvision': 'LayoutLMv3ImageProcessor'} (модель LayoutLMv3Config)
  • layoutxlm — {'torchvision': 'LayoutLMv2ImageProcessor', 'pil': 'LayoutLMv2ImageProcessorPil'} (модель LayoutXLMConfig)
  • levit — {'pil': 'LevitImageProcessorPil', 'torchvision': 'LevitImageProcessor'} (модель LevitConfig)
  • lfm2_vl — {'torchvision': 'Lfm2VlImageProcessor'} (модель Lfm2VlConfig)
  • lightglue — {'pil': 'LightGlueImageProcessorPil', 'torchvision': 'LightGlueImageProcessor'} (модель LightGlueConfig)
  • lighton_ocr — {'torchvision': 'PixtralImageProcessor', 'pil': 'PixtralImageProcessorPil'} (модель LightOnOcrConfig)
  • llama4 — {'torchvision': 'Llama4ImageProcessor'} (модель Llama4Config)
  • llava — {'pil': 'LlavaImageProcessorPil', 'torchvision': 'LlavaImageProcessor'} (модель LlavaConfig)
  • llava_next — {'pil': 'LlavaNextImageProcessorPil', 'torchvision': 'LlavaNextImageProcessor'} (модель LlavaNextConfig)
  • llava_next_video — {'torchvision': 'LlavaNextImageProcessor', 'pil': 'LlavaNextImageProcessorPil'} (модель LlavaNextVideoConfig)
  • llava_onevision — {'pil': 'LlavaOnevisionImageProcessorPil', 'torchvision': 'LlavaOnevisionImageProcessor'} (модель LlavaOnevisionConfig)
  • lw_detr — {'torchvision': 'DeformableDetrImageProcessor', 'pil': 'DeformableDetrImageProcessorPil'} (модель LwDetrConfig)
  • mask2former — {'pil': 'Mask2FormerImageProcessorPil', 'torchvision': 'Mask2FormerImageProcessor'} (модель Mask2FormerConfig)
  • maskformer — {'pil': 'MaskFormerImageProcessorPil', 'torchvision': 'MaskFormerImageProcessor'} (модель MaskFormerConfig)
  • metaclip_2 — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель MetaClip2Config)
  • mgp-str — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель MgpstrConfig)
  • minicpm_ocr — {'torchvision': 'MiniCPMOcrImageProcessor'} (модель MiniCPMOcrConfig)
  • minicpmv4_6 — {'pil': 'MiniCPMV4_6ImageProcessorPil', 'torchvision': 'MiniCPMV4_6ImageProcessor'} (модель MiniCPMV4_6Config)
  • minicpmv4_7 — {'pil': 'MiniCPMV4_6ImageProcessorPil', 'torchvision': 'MiniCPMV4_6ImageProcessor'} (модель MiniCPMV4_7Config)
  • minimax_m3_vl — {'torchvision': 'MiniMaxM3VLImageProcessor'} (модель MiniMaxM3VLConfig)
  • mistral3 — {'torchvision': 'PixtralImageProcessor', 'pil': 'PixtralImageProcessorPil'} (модель Mistral3Config)
  • mistral_large4 — {'torchvision': 'PixtralImageProcessor', 'pil': 'PixtralImageProcessorPil'} (модель MistralLarge4Config)
  • mlcd — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель MLCDVisionConfig)
  • mllama — {'pil': 'MllamaImageProcessorPil', 'torchvision': 'MllamaImageProcessor'} (модель MllamaConfig)
  • mm-grounding-dino — {'torchvision': 'GroundingDinoImageProcessor', 'pil': 'GroundingDinoImageProcessorPil'} (модель MMGroundingDinoConfig)
  • mobilenet_v1 — {'pil': 'MobileNetV1ImageProcessorPil', 'torchvision': 'MobileNetV1ImageProcessor'} (модель MobileNetV1Config)
  • mobilenet_v2 — {'pil': 'MobileNetV2ImageProcessorPil', 'torchvision': 'MobileNetV2ImageProcessor'} (модель MobileNetV2Config)
  • mobilevit — {'pil': 'MobileViTImageProcessorPil', 'torchvision': 'MobileViTImageProcessor'} (модель MobileViTConfig)
  • mobilevitv2 — {'torchvision': 'MobileViTImageProcessor', 'pil': 'MobileViTImageProcessorPil'} (модель MobileViTV2Config)
  • molmo2 — {'torchvision': 'Molmo2ImageProcessor'} (модель Molmo2Config)
  • muse_glimmer — {'torchvision': 'MuseGlimmerImageProcessor'} (модель MuseGlimmerConfig)
  • muse_spark — {'torchvision': 'MuseGlimmerImageProcessor'} (модель MuseSparkConfig)
  • nemotron_h_omni — {'torchvision': 'NemotronH_Omni_Reasoning_V3ImageProcessor'} (модель NemotronH_Omni_Reasoning_V3_Config)
  • neomme — {'torchvision': 'NeoMMEImageProcessor'} (модель NeoMMEConfig)
  • nougat — {'pil': 'NougatImageProcessorPil', 'torchvision': 'NougatImageProcessor'} (модель NougatConfig)
  • omdet-turbo — {'torchvision': 'DetrImageProcessor', 'pil': 'DetrImageProcessorPil'} (модель OmDetTurboConfig)
  • oneformer — {'pil': 'OneFormerImageProcessorPil', 'torchvision': 'OneFormerImageProcessor'} (модель OneFormerConfig)
  • ovis2 — {'pil': 'Ovis2ImageProcessorPil', 'torchvision': 'Ovis2ImageProcessor'} (модель Ovis2Config)
  • owlv2 — {'pil': 'Owlv2ImageProcessorPil', 'torchvision': 'Owlv2ImageProcessor'} (модель Owlv2Config)
  • owlvit — {'pil': 'OwlViTImageProcessorPil', 'torchvision': 'OwlViTImageProcessor'} (модель OwlViTConfig)
  • paddleocr_vl — {'pil': 'PaddleOCRVLImageProcessorPil', 'torchvision': 'PaddleOCRVLImageProcessor'} (модель PaddleOCRVLConfig)
  • paligemma — {'torchvision': 'SiglipImageProcessor', 'pil': 'SiglipImageProcessorPil'} (модель PaliGemmaConfig)
  • perceiver — {'pil': 'PerceiverImageProcessorPil', 'torchvision': 'PerceiverImageProcessor'} (модель PerceiverConfig)
  • perception_lm — {'torchvision': 'PerceptionLMImageProcessor'} (модель PerceptionLMConfig)
  • phi4_multimodal — {'torchvision': 'Phi4MultimodalImageProcessor'} (модель Phi4MultimodalConfig)
  • pi0 — {'torchvision': 'PI0ImageProcessor'} (модель PI0Config)
  • pix2struct — {'pil': 'Pix2StructImageProcessorPil', 'torchvision': 'Pix2StructImageProcessor'} (модель Pix2StructConfig)
  • pixio — {'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'} (модель PixioConfig)
  • pixtral — {'pil': 'PixtralImageProcessorPil', 'torchvision': 'PixtralImageProcessor'} (модель PixtralVisionConfig)
  • poolformer — {'pil': 'PoolFormerImageProcessorPil', 'torchvision': 'PoolFormerImageProcessor'} (модель PoolFormerConfig)
  • pp_chart2table — {'pil': 'PPChart2TableImageProcessorPil', 'torchvision': 'PPChart2TableImageProcessor'} (модель PPChart2TableConfig)
  • pp_doclayout_v2 — {'torchvision': 'PPDocLayoutV2ImageProcessor'} (модель PPDocLayoutV2Config)
  • pp_doclayout_v3 — {'torchvision': 'PPDocLayoutV3ImageProcessor'} (модель PPDocLayoutV3Config)
  • pp_doclayout_v4 — {'torchvision': 'PPDocLayoutV4ImageProcessor'} (модель PPDocLayoutV4Config)
  • pp_formulanet — {'torchvision': 'PPFormulaNetImageProcessor'} (модель PPFormulaNetConfig)
  • pp_lcnet — {'torchvision': 'PPLCNetImageProcessor'} (модель PPLCNetConfig)
  • pp_ocrv5_mobile_det — {'torchvision': 'PPOCRV5ServerDetImageProcessor'} (модель PPOCRV5MobileDetConfig)
  • pp_ocrv5_mobile_rec — {'torchvision': 'PPOCRV5ServerRecImageProcessor'} (модель PPOCRV5MobileRecConfig)
  • pp_ocrv5_server_det — {'torchvision': 'PPOCRV5ServerDetImageProcessor'} (модель PPOCRV5ServerDetConfig)
  • pp_ocrv5_server_rec — {'torchvision': 'PPOCRV5ServerRecImageProcessor'} (модель PPOCRV5ServerRecConfig)
  • pp_ocrv6_medium_det — {'torchvision': 'PPOCRV5ServerDetImageProcessor'} (модель PPOCRV6MediumDetConfig)
  • pp_ocrv6_small_det — {'torchvision': 'PPOCRV5ServerDetImageProcessor'} (модель PPOCRV6SmallDetConfig)
  • pp_ocrv6_small_rec — {'torchvision': 'PPOCRV6SmallRecImageProcessor'} (модель PPOCRV6SmallRecConfig)
  • pp_ocrv6_tiny_rec — {'torchvision': 'PPOCRV6SmallRecImageProcessor'} (модель PPOCRV6TinyRecConfig)
  • prompt_depth_anything — {'pil': 'PromptDepthAnythingImageProcessorPil', 'torchvision': 'PromptDepthAnythingImageProcessor'} (модель PromptDepthAnythingConfig)
  • pvt — {'pil': 'PvtImageProcessorPil', 'torchvision': 'PvtImageProcessor'} (модель PvtConfig)
  • pvt_v2 — {'torchvision': 'PvtImageProcessor', 'pil': 'PvtImageProcessorPil'} (модель PvtV2Config)
  • qianfan_ocr — {'torchvision': 'GotOcr2ImageProcessor', 'pil': 'GotOcr2ImageProcessorPil'} (модель QianfanOCRConfig)
  • qwen2_5_omni — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen2_5OmniConfig)
  • qwen2_5_vl — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen2_5_VLConfig)
  • qwen2_vl — {'pil': 'Qwen2VLImageProcessorPil', 'torchvision': 'Qwen2VLImageProcessor'} (модель Qwen2VLConfig)
  • qwen3_5 — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen3_5Config)
  • qwen3_5_moe — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen3_5MoeConfig)
  • qwen3_omni_moe — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen3OmniMoeConfig)
  • qwen3_vl — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen3VLConfig)
  • qwen4_exp — {'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'} (модель Qwen4ExpConfig)
  • regnet — {'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'} (модель RegNetConfig)
  • resnet — {'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'} (модель ResNetConfig)
  • rf_detr — {'torchvision': 'RfDetrImageProcessor'} (модель RfDetrConfig)
  • rt_detr — {'pil': 'RTDetrImageProcessorPil', 'torchvision': 'RTDetrImageProcessor'} (модель RTDetrConfig)
  • sam — {'pil': 'SamImageProcessorPil', 'torchvision': 'SamImageProcessor'} (модель SamConfig)
  • sam2 — {'torchvision': 'Sam2ImageProcessor'} (модель Sam2Config)
  • sam2_video — {'torchvision': 'Sam2ImageProcessor'} (модель Sam2VideoConfig)
  • sam3 — {'torchvision': 'Sam3ImageProcessor'} (модель Sam3Config)
  • sam3_lite_text — {'torchvision': 'Sam3ImageProcessor'} (модель Sam3LiteTextConfig)
  • sam3_tracker — {'torchvision': 'Sam3ImageProcessor'} (модель Sam3TrackerConfig)
  • sam3_tracker_video — {'torchvision': 'Sam3ImageProcessor'} (модель Sam3TrackerVideoConfig)
  • sam3_video — {'torchvision': 'Sam3ImageProcessor'} (модель Sam3VideoConfig)
  • sam_hq — {'torchvision': 'SamImageProcessor', 'pil': 'SamImageProcessorPil'} (модель SamHQConfig)
  • sapiens2 — {'torchvision': 'Sapiens2ImageProcessor'} (модель Sapiens2Config)
  • segformer — {'pil': 'SegformerImageProcessorPil', 'torchvision': 'SegformerImageProcessor'} (модель SegformerConfig)
  • seggpt — {'pil': 'SegGptImageProcessorPil', 'torchvision': 'SegGptImageProcessor'} (модель SegGptConfig)
  • shieldgemma2 — {'torchvision': 'Gemma3ImageProcessor', 'pil': 'Gemma3ImageProcessorPil'} (модель ShieldGemma2Config)
  • siglip — {'pil': 'SiglipImageProcessorPil', 'torchvision': 'SiglipImageProcessor'} (модель SiglipConfig)
  • siglip2 — {'pil': 'Siglip2ImageProcessorPil', 'torchvision': 'Siglip2ImageProcessor'} (модель Siglip2Config)
  • slanet — {'torchvision': 'SLANeXtImageProcessor'} (модель SLANetConfig)
  • slanext — {'torchvision': 'SLANeXtImageProcessor'} (модель SLANeXtConfig)
  • smolvlm — {'pil': 'SmolVLMImageProcessorPil', 'torchvision': 'SmolVLMImageProcessor'} (модель SmolVLMConfig)
  • step3p7 — {'torchvision': 'Step3p7ImageProcessor'} (модель Step3p7Config)
  • superglue — {'pil': 'SuperGlueImageProcessorPil', 'torchvision': 'SuperGlueImageProcessor'} (модель SuperGlueConfig)
  • superpoint — {'pil': 'SuperPointImageProcessorPil', 'torchvision': 'SuperPointImageProcessor'} (модель SuperPointConfig)
  • swiftformer — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель SwiftFormerConfig)
  • swin — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель SwinConfig)
  • swin2sr — {'pil': 'Swin2SRImageProcessorPil', 'torchvision': 'Swin2SRImageProcessor'} (модель Swin2SRConfig)
  • swinv2 — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель Swinv2Config)
  • t5gemma2 — {'torchvision': 'Gemma3ImageProcessor', 'pil': 'Gemma3ImageProcessorPil'} (модель T5Gemma2Config)
  • t5gemma2_encoder — {'torchvision': 'Gemma3ImageProcessor', 'pil': 'Gemma3ImageProcessorPil'} (модель T5Gemma2EncoderConfig)
  • table-transformer — {'torchvision': 'DetrImageProcessor', 'pil': 'DetrImageProcessorPil'} (модель TableTransformerConfig)
  • textnet — {'pil': 'TextNetImageProcessorPil', 'torchvision': 'TextNetImageProcessor'} (модель TextNetConfig)
  • timesformer — {'pil': 'VideoMAEImageProcessorPil', 'torchvision': 'VideoMAEImageProcessor'} (модель TimesformerConfig)
  • timm_wrapper — {'pil': 'TimmWrapperImageProcessor'} (модель TimmWrapperConfig)
  • tipsv2 — {'torchvision': 'Tipsv2ImageProcessor'} (модель Tipsv2Config)
  • tipsv2_dpt — {'torchvision': 'Tipsv2DptImageProcessor'} (модель Tipsv2DptConfig)
  • trocr — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель TrOCRConfig)
  • tvp — {'pil': 'TvpImageProcessorPil', 'torchvision': 'TvpImageProcessor'} (модель TvpConfig)
  • udop — {'torchvision': 'LayoutLMv3ImageProcessor', 'pil': 'LayoutLMv3ImageProcessorPil'} (модель UdopConfig)
  • ultravbert — {'torchvision': 'UltraVBertImageProcessor'} (модель UltraVBertConfig)
  • unlimited_ocr — {'torchvision': 'UnlimitedOcrImageProcessor'} (модель UnlimitedOcrConfig)
  • upernet — {'torchvision': 'SegformerImageProcessor', 'pil': 'SegformerImageProcessorPil'} (модель UperNetConfig)
  • uvdoc — {'torchvision': 'UVDocImageProcessor'} (модель UVDocConfig)
  • video_llama_3 — {'pil': 'VideoLlama3ImageProcessorPil', 'torchvision': 'VideoLlama3ImageProcessor'} (модель VideoLlama3Config)
  • video_llava — {'pil': 'VideoLlavaImageProcessor'} (модель VideoLlavaConfig)
  • videomae — {'pil': 'VideoMAEImageProcessorPil', 'torchvision': 'VideoMAEImageProcessor'} (модель VideoMAEConfig)
  • vilt — {'pil': 'ViltImageProcessorPil', 'torchvision': 'ViltImageProcessor'} (модель ViltConfig)
  • vipllava — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель VipLlavaConfig)
  • vit — {'pil': 'ViTImageProcessorPil', 'torchvision': 'ViTImageProcessor'} (модель ViTConfig)
  • vit_mae — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель ViTMAEConfig)
  • vit_msn — {'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'} (модель ViTMSNConfig)
  • vitmatte — {'pil': 'VitMatteImageProcessorPil', 'torchvision': 'VitMatteImageProcessor'} (модель VitMatteConfig)
  • vitpose — {'pil': 'VitPoseImageProcessorPil', 'torchvision': 'VitPoseImageProcessor'} (модель VitPoseConfig)
  • vivit — {'torchvision': 'VivitImageProcessor'} (модель VivitConfig)
  • xclip — {'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'} (модель XCLIPConfig)
  • yolos — {'pil': 'YolosImageProcessorPil', 'torchvision': 'YolosImageProcessor'} (модель YolosConfig)
  • zoedepth — {'pil': 'ZoeDepthImageProcessorPil', 'torchvision': 'ZoeDepthImageProcessor'} (модель ZoeDepthConfig)

Передача token=True обязательна, если вы хотите использовать частную модель.

Примеры:

>>> from transformers import AutoImageProcessor

>>> # Download image processor from huggingface.co and cache.
>>> image_processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")

>>> # If image processor files are in a directory (e.g. image processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # image_processor = AutoImageProcessor.from_pretrained("./test/saved_model/")

register

< >

( config_classslow_image_processor_class: type | None = Nonefast_image_processor_class: type | None = Noneimage_processor_classes: dict[str, type] | None = Noneexist_ok: bool = False )

Параметры

  • config_class (PreTrainedConfig) — Конфигурация, соответствующая модели, которую нужно зарегистрировать.
  • slow_image_processor_class (type, необязательный) — Класс обработчика изображений с бэкендом PIL (устаревший, используйте image_processor_classes={"pil": ...}).
  • fast_image_processor_class (type, необязательный) — Класс процессора изображений для бэкенда Torchvision (устаревший, используйте image_processor_classes={"torchvision": ...}).
  • image_processor_classes (dict[str, type], необязательный) — Словарь, сопоставляющий имена бэкендов с классами процессоров изображений. Позволяет регистрировать пользовательские бэкенды. Пример: {"pil": MyPilProcessor, "torchvision": MyTorchvisionProcessor, "custom": MyCustomProcessor}
  • exist_ok (bool, необязательный, по умолчанию False) — Если True, разрешает перезаписывать существующие регистрации.

Зарегистрируйте новый обработчик изображений для этого класса.

AutoVideoProcessor

См. Бэкенды видеопроцессора для выбора бэкенда, пользовательской регистрации, а также сохранения и повторной загрузки.

класс transformers.AutoVideoProcessor

< >

( )

Это универсальный класс видеопроцессора, который при создании с помощью метода класса AutoVideoProcessor.from_pretrained() будет инстанцирован как один из классов видеопроцессоров библиотеки.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_pretrained

< >

( pretrained_model_name_or_path*inputs**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Это может быть:
    • строка, model id предобученного video_processor, размещённого в репозитории модели на huggingface.co.
    • путь к каталогу, содержащему файл видео-процессора, сохранённый с помощью метода save_pretrained(), например, ./my_model_directory/.
    • путь к сохранённому JSON-файлу видео-процессора, например, ./my_model_directory/preprocessor_config.json.
  • cache_dir (str or os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженный предобученный видеопроцессор модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Указывает, следует ли принудительно (пере)загружать файлы видео-процессора и заменять кэшированные версии, если они уже существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • token (str или bool, необязательный) — Токен, используемый в качестве HTTP bearer-авторизации для удалённых файлов. Если True, будет использован токен, сгенерированный при выполнении hf auth login (хранится в ~/.huggingface).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • return_unused_kwargs (bool, необязательный, по умолчанию False) — Если False, функция возвращает только итоговый объект видеопроцессора. Если True, функция возвращает Tuple(video_processor, unused_kwargs), где unused_kwargs — это словарь, содержащий пары ключ/значение, ключи которых не являются атрибутами видеопроцессора, то есть та часть kwargs, которая не была использована для обновления video_processor и в остальном игнорируется.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • kwargs (dict[str, Any], необязательный) — Значения в kwargs для любых ключей, являющихся атрибутами видео-процессора, используются для переопределения загруженных значений. Поведение в отношении пар ключ/значение, ключи которых не являются атрибутами видео-процессора, контролируется параметром return_unused_kwargs.

Создайте экземпляр одного из классов видео-процессоров библиотеки на основе словаря предобученной модели.

Класс видеопроцессора, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

  • cohere_compass — {'torchvision': 'CohereCompassVideoProcessor'} (модель CohereCompassConfig)
  • cosmos3_edge — {'torchvision': 'Cosmos3EdgeVideoProcessor'} (модель Cosmos3EdgeConfig)
  • cosmos3_omni — {'torchvision': 'Qwen3VLVideoProcessor'} (модель Cosmos3OmniConfig)
  • embedding_gemma2 — {'torchvision': 'EmbeddingGemma2VideoProcessor'} (модель EmbeddingGemma2Config)
  • ernie4_5_vl_moe — {'torchvision': 'Ernie4_5_VLMoeVideoProcessor'} (модель Ernie4_5_VLMoeConfig)
  • exaone4_5 — {'torchvision': 'Qwen2VLVideoProcessor'} (модель Exaone4_5_Config)
  • gemma4 — {'torchvision': 'Gemma4VideoProcessor'} (модель Gemma4Config)
  • gemma4_unified — {'torchvision': 'Gemma4UnifiedVideoProcessor'} (модель Gemma4UnifiedConfig)
  • glm46v — {'torchvision': 'Glm46VVideoProcessor'} (модель Glm46VConfig)
  • glm4v — {'torchvision': 'Glm4vVideoProcessor'} (модель Glm4vConfig)
  • glm5_next — {'torchvision': 'Glm5NextVideoProcessor'} (модель Glm5NextConfig)
  • glmga — {'torchvision': 'GlmgaVideoProcessor'} (модель GlmgaConfig)
  • hyperclovax_vision_v2 — {'torchvision': 'Qwen2VLVideoProcessor'} (модель HyperCLOVAXVisionV2Config)
  • instructblip — {'torchvision': 'InstructBlipVideoVideoProcessor'} (модель InstructBlipConfig)
  • instructblipvideo — {'torchvision': 'InstructBlipVideoVideoProcessor'} (модель InstructBlipVideoConfig)
  • internvl — {'torchvision': 'InternVLVideoProcessor'} (модель InternVLConfig)
  • kimi_k25 — {'torchvision': 'Kimi_K25VideoProcessor'} (модель Kimi_K25Config)
  • llava_next_video — {'torchvision': 'LlavaNextVideoVideoProcessor'} (модель LlavaNextVideoConfig)
  • llava_onevision — {'torchvision': 'LlavaOnevisionVideoProcessor'} (модель LlavaOnevisionConfig)
  • minicpm_ocr — {'torchvision': 'MiniCPMOcrVideoProcessor'} (модель MiniCPMOcrConfig)
  • minicpmv4_6 — {'torchvision': 'MiniCPMV4_6VideoProcessor'} (модель MiniCPMV4_6Config)
  • minicpmv4_7 — {'torchvision': 'MiniCPMV4_6VideoProcessor'} (модель MiniCPMV4_7Config)
  • minimax_m3_vl — {'torchvision': 'MiniMaxM3VLVideoProcessor'} (модель MiniMaxM3VLConfig)
  • molmo2 — {'torchvision': 'Molmo2VideoProcessor'} (модель Molmo2Config)
  • muse_glimmer — {'torchvision': 'MuseGlimmerVideoProcessor'} (модель MuseGlimmerConfig)
  • muse_spark — {'torchvision': 'MuseGlimmerVideoProcessor'} (модель MuseSparkConfig)
  • nemotron_h_omni — {'torchvision': 'NemotronH_Omni_Reasoning_V3VideoProcessor'} (модель NemotronH_Omni_Reasoning_V3_Config)
  • pe_audio_video — {'torchvision': 'PeVideoVideoProcessor'} (модель PeAudioVideoConfig)
  • pe_video — {'torchvision': 'PeVideoVideoProcessor'} (модель PeVideoConfig)
  • perception_lm — {'torchvision': 'PerceptionLMVideoProcessor'} (модель PerceptionLMConfig)
  • qwen2_5_omni — {'torchvision': 'Qwen2VLVideoProcessor'} (модель Qwen2_5OmniConfig)
  • qwen2_5_vl — {'torchvision': 'Qwen2VLVideoProcessor'} (модель Qwen2_5_VLConfig)
  • qwen2_vl — {'torchvision': 'Qwen2VLVideoProcessor'} (модель Qwen2VLConfig)
  • qwen3_5 — {'torchvision': 'Qwen3VLVideoProcessor'} (модель Qwen3_5Config)
  • qwen3_5_moe — {'torchvision': 'Qwen3VLVideoProcessor'} (модель Qwen3_5MoeConfig)
  • qwen3_omni_moe — {'torchvision': 'Qwen2VLVideoProcessor'} (модель Qwen3OmniMoeConfig)
  • qwen3_vl — {'torchvision': 'Qwen3VLVideoProcessor'} (модель Qwen3VLConfig)
  • qwen3_vl_moe — {'torchvision': 'Qwen3VLVideoProcessor'} (модель Qwen3VLMoeConfig)
  • qwen4_exp — {'torchvision': 'Qwen3VLVideoProcessor'} (модель Qwen4ExpConfig)
  • sam2_video — {'torchvision': 'Sam2VideoVideoProcessor'} (модель Sam2VideoConfig)
  • smolvlm — {'torchvision': 'SmolVLMVideoProcessor'} (модель SmolVLMConfig)
  • timesformer — {'torchvision': 'TimesformerVideoProcessor'} (модель TimesformerConfig)
  • video_llama_3 — {'torchvision': 'VideoLlama3VideoProcessor'} (модель VideoLlama3Config)
  • video_llava — {'torchvision': 'VideoLlavaVideoProcessor'} (модель VideoLlavaConfig)
  • videomae — {'torchvision': 'VideoMAEVideoProcessor'} (модель VideoMAEConfig)
  • videomt — {'torchvision': 'VideomtVideoProcessor'} (модель VideomtConfig)
  • videoprism — {'torchvision': 'LlavaOnevisionVideoProcessor'} (модель VideoPrismConfig)
  • vivit — {'torchvision': 'VivitVideoProcessor'} (модель VivitConfig)
  • vjepa2 — {'torchvision': 'VJEPA2VideoProcessor'} (модель VJEPA2Config)

Передача token=True обязательна, если вы хотите использовать частную модель.

Примеры:

>>> from transformers import AutoVideoProcessor

>>> # Download video processor from huggingface.co and cache.
>>> video_processor = AutoVideoProcessor.from_pretrained("llava-hf/llava-onevision-qwen2-0.5b-ov-hf")

>>> # If video processor files are in a directory (e.g. video processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # video_processor = AutoVideoProcessor.from_pretrained("./test/saved_model/")

register

< >

( config_classvideo_processor_class: type | None = Nonevideo_processor_classes: dict[str, type] | None = Noneexist_ok: bool = False )

Параметры

  • config_class (PreTrainedConfig) — Конфигурация, соответствующая модели, которую нужно зарегистрировать.
  • video_processor_class (BaseVideoProcessor) — Класс видео-процессора, который нужно зарегистрировать.
  • video_processor_classes (dict[str, type], необязательный) — Словарь, сопоставляющий имена бэкендов с классами процессоров видео. Позволяет регистрировать пользовательские бэкенды. Пример: {"pil": MyPilVideoProcessor, "torchvision": MyTorchvisionVideoProcessor, "custom": MyCustomVideoProcessor}
  • exist_ok (bool, необязательный, по умолчанию False) — Если True, разрешает перезаписывать существующие регистрации.

Зарегистрируйте новый видео-процессор для этого класса.

AutoProcessor

класс transformers.AutoProcessor

< >

( )

Это универсальный класс процессора, который при создании с помощью метода класса AutoProcessor.from_pretrained() будет инстанцирован как один из классов процессоров библиотеки.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_pretrained

< >

( pretrained_model_name_or_path**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Это может быть:
    • строка, model id предобученного feature_extractor, размещенного в репозитории модели на huggingface.co.
    • путь к каталогу, содержащему файлы процессора, сохранённые с помощью метода save_pretrained(), например, ./my_model_directory/.
  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженный предобученный экстрактор признаков модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать файлы экстрактора признаков и заменять кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • token (str или bool, необязательный) — Токен, используемый в качестве HTTP bearer-авторизации для удалённых файлов. Если True, будет использован токен, сгенерированный при выполнении hf auth login (хранится в ~/.huggingface).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • return_unused_kwargs (bool, необязательный, по умолчанию False) — Если False, функция возвращает только итоговый объект feature extractor. Если True, функция возвращает Tuple(feature_extractor, unused_kwargs), где unused_kwargs — это словарь, содержащий пары ключ/значение, ключи которых не являются атрибутами feature extractor, то есть та часть kwargs, которая не была использована для обновления feature_extractor и в остальном игнорируется.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • kwargs (dict[str, Any], необязательный) — Значения в kwargs для любых ключей, являющихся атрибутами feature extractor, будут использоваться для переопределения загруженных значений. Поведение в отношении пар ключ/значение, ключи которых не являются атрибутами feature extractor, контролируется параметром return_unused_kwargs.

Создайте экземпляр одного из классов процессоров библиотеки на основе словаря предобученной модели.

Класс процессора, который будет создан, выбирается на основе свойства model_type объекта конфигурации (переданного в качестве аргумента или загруженного из pretrained_model_name_or_path, если это возможно):

Передача token=True обязательна, если вы хотите использовать частную модель.

Примеры:

>>> from transformers import AutoProcessor

>>> # Download processor from huggingface.co and cache.
>>> processor = AutoProcessor.from_pretrained("facebook/wav2vec2-base-960h")

>>> # If processor files are in a directory (e.g. processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # processor = AutoProcessor.from_pretrained("./test/saved_model/")

register

< >

( config_classprocessor_classexist_ok = False )

Параметры

  • config_class (PreTrainedConfig) — Конфигурация, соответствующая модели, которую нужно зарегистрировать.
  • processor_class (ProcessorMixin) — процессор для регистрации.

Зарегистрируйте новый процессор для этого класса.

Универсальные классы моделей

Для создания базового класса модели без конкретного слоя-головки доступны следующие автоклассы:

AutoModel

класс transformers.AutoModel

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из базовых классов моделей библиотеки.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из базовых классов моделей библиотеки на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModel

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModel.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из базовых классов моделей библиотеки из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModel

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModel.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModel.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

Универсальные классы предварительного обучения

Для создания модели с головой предобучения доступны следующие автоклассы.

AutoModelForPreTraining

класс transformers.AutoModelForPreTraining

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с заголовком для предобучения).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой предобучения) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForPreTraining

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForPreTraining.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой предобучения) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForPreTraining

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForPreTraining.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForPreTraining.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

Обработка естественного языка

Для следующих задач обработки естественного языка доступны следующие автоклассы.

AutoModelForCausalLM

класс transformers.AutoModelForCausalLM

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для каузального языкового моделирования).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для каузального моделирования языка) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForCausalLM

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForCausalLM.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с заголовком для каузального языкового моделирования) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForCausalLM

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForCausalLM.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForCausalLM.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForMaskedLM

класс transformers.AutoModelForMaskedLM

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для маскированного языкового моделирования).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для маскированного языкового моделирования) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForMaskedLM

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForMaskedLM.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для маскированного языкового моделирования) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForMaskedLM

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForMaskGeneration

класс transformers.AutoModelForMaskGeneration

< >

( *args**kwargs )

AutoModelForSeq2SeqLM

класс transformers.AutoModelForSeq2SeqLM

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для последовательностно-последовательностного языкового моделирования).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для последовательностно-последовательностного языкового моделирования) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForSeq2SeqLM

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-t5/t5-base")
>>> model = AutoModelForSeq2SeqLM.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для последовательностно-последовательностного языкового моделирования) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForSeq2SeqLM

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-base")

>>> # Update configuration during loading
>>> model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-base", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForSequenceClassification

класс transformers.AutoModelForSequenceClassification

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для классификации последовательностей).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для классификации последовательностей) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForSequenceClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForSequenceClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для классификации последовательностей) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForSequenceClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSequenceClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForSequenceClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForMultipleChoice

класс transformers.AutoModelForMultipleChoice

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для множественного выбора).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для задачи множественного выбора) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForMultipleChoice

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForMultipleChoice.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для задачи выбора из нескольких вариантов) на основе предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForMultipleChoice

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMultipleChoice.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForMultipleChoice.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForNextSentencePrediction

класс transformers.AutoModelForNextSentencePrediction

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для предсказания следующего предложения).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для предсказания следующего предложения) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForNextSentencePrediction

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForNextSentencePrediction.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для предсказания следующего предложения) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForNextSentencePrediction

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForNextSentencePrediction.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForNextSentencePrediction.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTokenClassification

класс transformers.AutoModelForTokenClassification

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для классификации токенов).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для классификации токенов) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForTokenClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForTokenClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для классификации токенов) на основе предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForTokenClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTokenClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForTokenClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForQuestionAnswering

класс transformers.AutoModelForQuestionAnswering

< >

( *args**kwargs )

This is a generic model class that will be instantiated as one of the model classes of the library (with a question answering head) when created with the from_pretrained() class method or the from_config() class method.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для задачи вопрос-ответ) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForQuestionAnswering

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForQuestionAnswering.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для задачи question answering) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForQuestionAnswering

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForQuestionAnswering.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForQuestionAnswering.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTextEncoding

класс transformers.AutoModelForTextEncoding

< >

( *args**kwargs )

Компьютерное зрение

Для следующих задач компьютерного зрения доступны следующие автоклассы.

AutoModelForDepthEstimation

класс transformers.AutoModelForDepthEstimation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для оценки глубины).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для оценки глубины) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForDepthEstimation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForDepthEstimation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для оценки глубины) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForDepthEstimation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForDepthEstimation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForDepthEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForNormalEstimation

класс transformers.AutoModelForNormalEstimation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с обычной оценочной головой).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с обычной оценочной головой) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForNormalEstimation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForNormalEstimation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с обычной оценочной головой) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForNormalEstimation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForNormalEstimation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForNormalEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForPointmapEstimation

класс transformers.AutoModelForPointmapEstimation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для оценки pointmap).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для оценки pointmap) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForPointmapEstimation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForPointmapEstimation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для оценки pointmap) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForPointmapEstimation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForPointmapEstimation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForPointmapEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForImageMatting

класс transformers.AutoModelForImageMatting

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для маттинга изображений).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт экземпляр одного из классов моделей библиотеки (с головой для маттинга изображений) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageMatting

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForImageMatting.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для маттинга изображений) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageMatting

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageMatting.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForImageMatting.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTextRecognition

класс transformers.AutoModelForTextRecognition

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для распознавания текста).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для распознавания текста) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForTextRecognition

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForTextRecognition.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для распознавания текста) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForTextRecognition

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTextRecognition.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForTextRecognition.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTableRecognition

класс transformers.AutoModelForTableRecognition

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой распознавания таблиц).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для распознавания таблиц) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForTableRecognition

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForTableRecognition.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для распознавания таблиц) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForTableRecognition

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTableRecognition.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForTableRecognition.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForImageClassification

класс transformers.AutoModelForImageClassification

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для классификации изображений).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для классификации изображений) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForImageClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для классификации изображений) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForImageClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForVideoClassification

класс transformers.AutoModelForVideoClassification

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для классификации видео).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для классификации видео) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForVideoClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForVideoClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для классификации видео) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForVideoClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForVideoClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForVideoClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForPoseEstimation

класс transformers.AutoModelForPoseEstimation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для оценки позы).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для оценки позы) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForPoseEstimation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForPoseEstimation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для оценки позы) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForPoseEstimation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForPoseEstimation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForPoseEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForKeypointDetection

класс transformers.AutoModelForKeypointDetection

< >

( *args**kwargs )

AutoModelForKeypointMatching

класс transformers.AutoModelForKeypointMatching

< >

( *args**kwargs )

AutoModelForMaskedImageModeling

класс transformers.AutoModelForMaskedImageModeling

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для моделирования изображений с маскированием).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для моделирования маскированных изображений) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForMaskedImageModeling

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForMaskedImageModeling.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для маскированного моделирования изображений) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForMaskedImageModeling

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMaskedImageModeling.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForMaskedImageModeling.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForObjectDetection

класс transformers.AutoModelForObjectDetection

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для обнаружения объектов).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для обнаружения объектов) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForObjectDetection

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForObjectDetection.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для обнаружения объектов) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForObjectDetection

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForObjectDetection.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForObjectDetection.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForImageSegmentation

класс transformers.AutoModelForImageSegmentation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для сегментации изображений).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт экземпляр одного из классов моделей библиотеки (с головой для сегментации изображений) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageSegmentation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForImageSegmentation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для сегментации изображений) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageSegmentation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageSegmentation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForImageSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForImageToImage

класс transformers.AutoModelForImageToImage

< >

( *args**kwargs )

AutoModelForSemanticSegmentation

класс transformers.AutoModelForSemanticSegmentation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой семантической сегментации).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой семантической сегментации) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForSemanticSegmentation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForSemanticSegmentation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой семантической сегментации) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForSemanticSegmentation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSemanticSegmentation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForSemanticSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForInstanceSegmentation

класс transformers.AutoModelForInstanceSegmentation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов модели библиотеки (с головой для сегментации экземпляров).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для сегментации экземпляров) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForInstanceSegmentation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForInstanceSegmentation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для сегментации экземпляров) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForInstanceSegmentation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForInstanceSegmentation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForInstanceSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForUniversalSegmentation

класс transformers.AutoModelForUniversalSegmentation

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с универсальной головой сегментации изображений).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с универсальной головой сегментации изображений) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForUniversalSegmentation

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForUniversalSegmentation.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с универсальной головой сегментации изображений) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForUniversalSegmentation

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForUniversalSegmentation.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForUniversalSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForZeroShotImageClassification

класс transformers.AutoModelForZeroShotImageClassification

< >

( *args**kwargs )

This is a generic model class that will be instantiated as one of the model classes of the library (with a zero-shot image classification head) when created with the from_pretrained() class method or the from_config() class method.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с заголовком для классификации изображений в режиме zero-shot) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForZeroShotImageClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForZeroShotImageClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головкой для классификации изображений в режиме zero-shot) на основе предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForZeroShotImageClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForZeroShotImageClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForZeroShotImageClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForZeroShotObjectDetection

класс transformers.AutoModelForZeroShotObjectDetection

< >

( *args**kwargs )

This is a generic model class that will be instantiated as one of the model classes of the library (with a zero-shot object detection head) when created with the from_pretrained() class method or the from_config() class method.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головкой для детекции объектов в zero-shot режиме) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForZeroShotObjectDetection

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForZeroShotObjectDetection.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для детекции объектов в zero-shot режиме) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForZeroShotObjectDetection

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForZeroShotObjectDetection.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForZeroShotObjectDetection.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

Аудио

Для следующих задач обработки аудио доступны следующие автоклассы.

AutoModelForAudioClassification

класс transformers.AutoModelForAudioClassification

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для классификации аудио).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для классификации аудио) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForAudioClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для классификации аудио) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForAudioClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForAudioFrameClassification

класс transformers.AutoModelForAudioFrameClassification

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой классификации аудиокадров (токенов)).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой классификации аудиокадров (токенов)) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioFrameClassification

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForAudioFrameClassification.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой классификации аудиокадров (токенов)) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioFrameClassification

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioFrameClassification.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForAudioFrameClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForCTC

класс transformers.AutoModelForCTC

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для связного временного классификационного обучения).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с головой для классификации с временной связностью) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForCTC

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForCTC.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой connectionist temporal classification) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForCTC

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForCTC.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForCTC.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTDT

класс transformers.AutoModelForTDT

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой transducer для токенов и длительности).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт экземпляр одного из классов моделей библиотеки (с головой transducer для токенов и длительности) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForTDT

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForTDT.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой transducer, работающей с токенами и длительностью) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForTDT

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTDT.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForTDT.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForRNNT

класс transformers.AutoModelForRNNT

< >

( *args**kwargs )

This is a generic model class that will be instantiated as one of the model classes of the library (with a RNN transducer head) when created with the from_pretrained() class method or the from_config() class method.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт экземпляр одного из классов моделей библиотеки (с головой RNN-транскрибера) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForRNNT

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForRNNT.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой RNN-транскрибтера) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForRNNT

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForRNNT.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForRNNT.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForSpeechSeq2Seq

класс transformers.AutoModelForSpeechSeq2Seq

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой моделирования «последовательность в последовательность» для распознавания речи).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с заголовком моделирования «последовательность в последовательность» для распознавания речи) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForSpeechSeq2Seq

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForSpeechSeq2Seq.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с заголовком моделирования «последовательность в последовательность» для распознавания речи) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForSpeechSeq2Seq

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSpeechSeq2Seq.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForSpeechSeq2Seq.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForAudioXVector

класс transformers.AutoModelForAudioXVector

< >

( *args**kwargs )

This is a generic model class that will be instantiated as one of the model classes of the library (with a audio retrieval via x-vector head) when created with the from_pretrained() class method or the from_config() class method.

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с аудио-ретривалом через x-vector head) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioXVector

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForAudioXVector.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с аудио-поиском через x-vector head) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioXVector

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioXVector.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForAudioXVector.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTextToSpectrogram

класс transformers.AutoModelForTextToSpectrogram

< >

( *args**kwargs )

AutoModelForTextToWaveform

класс transformers.AutoModelForTextToWaveform

< >

( *args**kwargs )

AutoModelForAudioTokenization

класс transformers.AutoModelForAudioTokenization

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с аудио-токенизацией через кодбуки).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с аудио-токенизацией через кодбуки) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioTokenization

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForAudioTokenization.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с аудио-токенизацией через кодбуки) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForAudioTokenization

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioTokenization.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForAudioTokenization.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

Мультимодальность

Для следующих мультимодальных задач доступны следующие автоклассы.

AutoModelForMultimodalLM

класс transformers.AutoModelForMultimodalLM

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с мультимодальной генеративной головой).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с мультимодальной генеративной головой) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForMultimodalLM

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForMultimodalLM.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с мультимодальной генерационной головой) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForMultimodalLM

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMultimodalLM.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForMultimodalLM.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForTableQuestionAnswering

класс transformers.AutoModelForTableQuestionAnswering

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для ответа на вопросы по тексту).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для задачи вопросно-ответной) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForTableQuestionAnswering

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google/tapas-base-finetuned-wtq")
>>> model = AutoModelForTableQuestionAnswering.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для табличного вопросно-ответного) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForTableQuestionAnswering

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTableQuestionAnswering.from_pretrained("google/tapas-base-finetuned-wtq")

>>> # Update configuration during loading
>>> model = AutoModelForTableQuestionAnswering.from_pretrained("google/tapas-base-finetuned-wtq", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForDocumentQuestionAnswering

класс transformers.AutoModelForDocumentQuestionAnswering

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для ответа на вопросы по документу).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт один из классов моделей библиотеки (с головой для задачи вопросно-ответной работы с документами) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForDocumentQuestionAnswering

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("impira/layoutlm-document-qa", revision="52e01b3")
>>> model = AutoModelForDocumentQuestionAnswering.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для задачи вопросно-ответной работы с документами) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForDocumentQuestionAnswering

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForDocumentQuestionAnswering.from_pretrained("impira/layoutlm-document-qa", revision="52e01b3")

>>> # Update configuration during loading
>>> model = AutoModelForDocumentQuestionAnswering.from_pretrained("impira/layoutlm-document-qa", revision="52e01b3", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForVisualQuestionAnswering

класс transformers.AutoModelForVisualQuestionAnswering

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для визуального вопросно-ответного моделирования).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для визуального вопросно-ответного моделирования) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForVisualQuestionAnswering

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
>>> model = AutoModelForVisualQuestionAnswering.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для визуального вопросно-ответного моделирования) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForVisualQuestionAnswering

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForVisualQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")

>>> # Update configuration during loading
>>> model = AutoModelForVisualQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa", output_attentions=True)
>>> model.config.output_attentions
True

AutoModelForImageTextToText

класс transformers.AutoModelForImageTextToText

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с заголовком моделирования «изображение-текст в текст»).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

Создаёт экземпляр одного из классов моделей библиотеки (с заголовком моделирования «изображение-текст в текст») на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageTextToText

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForImageTextToText.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с заголовком моделирования «изображение-текст в текст») на основе предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForImageTextToText

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageTextToText.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForImageTextToText.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True

Временные ряды

AutoModelForTimeSeriesPrediction

класс transformers.AutoModelForTimeSeriesPrediction

< >

( *args**kwargs )

Это универсальный класс модели, который при создании с помощью метода класса from_pretrained() или метода класса from_config() будет инстанцирован как один из классов моделей библиотеки (с головой для прогнозирования временных рядов).

Этот класс нельзя создать напрямую с помощью __init__() (выбрасывает ошибку).

from_config

< >

( **kwargs )

Параметры

  • config (PreTrainedConfig) — Класс модели, который будет создан, определяется на основе класса конфигурации:
  • attn_implementation (str, необязательный) — Реализация механизма внимания, используемая в модели (если применимо). Допустимые значения:

Создаёт один из классов моделей библиотеки (с головой для прогнозирования временных рядов) на основе конфигурации.

Примечание: Загрузка модели из файла конфигурации не загружает веса модели. Это влияет только на конфигурацию модели. Используйте from_pretrained() для загрузки весов модели.

Примеры:

>>> from transformers import AutoConfig, AutoModelForTimeSeriesPrediction

>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-cased")
>>> model = AutoModelForTimeSeriesPrediction.from_config(config)

from_pretrained

< >

( *model_args**kwargs )

Параметры

  • pretrained_model_name_or_path (str или os.PathLike) — Может быть:
    • Строка, model id предобученной модели, размещенной в репозитории моделей на huggingface.co.
    • Путь к каталогу, содержащему веса модели, сохранённые с помощью save_pretrained(), например, ./my_model_directory/.
  • model_args (дополнительные позиционные аргументы, необязательные) — будут переданы в метод __init__() базовой модели.
  • config (PreTrainedConfig, необязательный) — Конфигурация модели, используемая вместо автоматически загружаемой. Конфигурация может быть загружена автоматически, если:
    • Модель — это модель, предоставляемая библиотекой (загружаемая с помощью строки model id предобученной модели).
    • Модель была сохранена с помощью save_pretrained() и загружается заново путем указания каталога сохранения.
    • Модель загружается путём указания локального каталога в качестве pretrained_model_name_or_path, при этом в каталоге находится файл конфигурации JSON с именем config.json.
  • state_dict (dict[str, torch.Tensor], необязательный) — Словарь состояния, который используется вместо словаря состояния, загруженного из файла сохранённых весов. Этот параметр можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым вариантом.

    Этот вариант можно использовать, если вы хотите создать модель на основе предобученной конфигурации, но загрузить собственные веса. В этом случае, однако, стоит проверить, не является ли использование save_pretrained() и from_pretrained() более простым решением.

  • cache_dir (str или os.PathLike, необязательный) — Путь к каталогу, в котором следует кэшировать загруженную конфигурацию предобученной модели, если стандартный кэш не используется.
  • force_download (bool, необязательный, по умолчанию False) — Следует ли принудительно (пере)скачивать веса модели и файлы конфигурации, игнорируя кэшированные версии, если они существуют.
  • proxies (dict[str, str], необязательный) — Словарь прокси-серверов, используемых по протоколу или конечной точке, например, {'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Прокси применяются при каждом запросе.
  • output_loading_info(bool, optional, по умолчанию False) — Определяет, следует ли также возвращать словарь, содержащий отсутствующие ключи, неожиданные ключи и сообщения об ошибках.
  • local_files_only(bool, необязательный, по умолчанию False) — Определяет, следует ли использовать только локальные файлы (например, не пытаться загрузить модель).
  • revision (str, необязательный, по умолчанию "main") — Конкретная версия модели, которую следует использовать. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • trust_remote_code (bool, необязательный, по умолчанию False) — Определяет, разрешено ли использование пользовательских моделей, определенных на Hub в собственных файлах моделирования. Этот параметр следует устанавливать в True только для репозиториев, которым вы доверяете и код которых вы прочитали, поскольку он будет выполнять код, присутствующий на Hub, на вашем локальном компьютере.
  • code_revision (str, необязательный, по умолчанию "main") — Конкретная ревизия, используемая для кода на Hub, если код находится в другом репозитории, чем остальная часть модели. Это может быть имя ветки, имя тега или идентификатор коммита, поскольку для хранения моделей и других артефактов на huggingface.co используется система на базе git, поэтому revision может быть любым идентификатором, разрешённым git.
  • kwargs (дополнительные именованные аргументы, необязательный) — Можно использовать для обновления объекта конфигурации (после его загрузки) и инициализации модели (например, output_attentions=True). Поведение зависит от того, предоставляется ли config или загружается автоматически:
    • Если конфигурация передана с помощью config, **kwargs будут непосредственно переданы в метод __init__ базовой модели (мы предполагаем, что все необходимые обновления конфигурации уже были выполнены)
    • Если конфигурация не предоставлена, kwargs сначала передаются в функцию инициализации класса конфигурации (from_pretrained()). Каждое ключевое значение kwargs, соответствующее атрибуту конфигурации, используется для переопределения этого атрибута переданным значением из kwargs. Оставшиеся ключи, не соответствующие ни одному атрибуту конфигурации, передаются в функцию __init__ базовой модели.

Создайте экземпляр одного из классов моделей библиотеки (с головой для прогнозирования временных рядов) из предобученной модели.

Класс модели, который будет создан, выбирается на основе свойства model_type объекта конфигурации (передаваемого в качестве аргумента или загружаемого из pretrained_model_name_or_path, если это возможно), а при его отсутствии — с помощью сопоставления по шаблону для pretrained_model_name_or_path:

По умолчанию модель переводится в режим оценки с помощью model.eval() (например, модули dropout отключаются). Для обучения модели сначала необходимо вернуть её в режим обучения с помощью model.train()

Примеры:

>>> from transformers import AutoConfig, AutoModelForTimeSeriesPrediction

>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTimeSeriesPrediction.from_pretrained("google-bert/bert-base-cased")

>>> # Update configuration during loading
>>> model = AutoModelForTimeSeriesPrediction.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True
Update on GitHub