Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions docs/source/ko/_toctree.yml
Original file line number Diff line number Diff line change
Expand Up @@ -740,8 +740,8 @@
title: (๋ฒˆ์—ญ์ค‘) TVLT
- local: in_translation
title: (๋ฒˆ์—ญ์ค‘) ViLT
- local: in_translation
title: (๋ฒˆ์—ญ์ค‘) Vision Encoder Decoder Models
- local: model_doc/vision-encoder-decoder
title: ๋น„์ „ ์ธ์ฝ”๋”-๋””์ฝ”๋” ๋ชจ๋ธ
- local: in_translation
title: (๋ฒˆ์—ญ์ค‘) Vision Text Dual Encoder
- local: in_translation
Expand Down
184 changes: 184 additions & 0 deletions docs/source/ko/model_doc/vision-encoder-decoder.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,184 @@
<!--Copyright 2021 The HuggingFace Team. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
the License. You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.

โš ๏ธ Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be
rendered properly in your Markdown viewer.

-->

# ๋น„์ „ ์ธ์ฝ”๋”-๋””์ฝ”๋” ๋ชจ๋ธ[[vision-encoder-decoder-models]]

<div class="flex flex-wrap space-x-1">
<img alt="PyTorch" src="https://img.shields.io/badge/PyTorch-DE3412?style=flat&logo=pytorch&logoColor=white">
<img alt="TensorFlow" src="https://img.shields.io/badge/TensorFlow-FF6F00?style=flat&logo=tensorflow&logoColor=white">
<img alt="Flax" src="https://img.shields.io/badge/Flax-29a79b.svg?style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAC0AAAAtCAMAAAANxBKoAAAC7lBMVEUAAADg5vYHPVgAoJH+/v76+v39/f9JbLP///9+AIgAnY3///+mcqzt8fXy9fgkXa3Ax9709fr+///9/f8qXq49qp5AaLGMwrv8/P0eW60VWawxYq8yqJzG2dytt9Wyu9elzci519Lf3O3S2efY3OrY0+Xp7PT///////+dqNCexMc6Z7AGpJeGvbenstPZ5ejQ1OfJzOLa7ejh4+/r8fT29vpccbklWK8PVa0AS6ghW63O498vYa+lsdKz1NDRt9Kw1c672tbD3tnAxt7R6OHp5vDe7OrDyuDn6vLl6/EAQKak0MgATakkppo3ZK/Bz9y8w9yzu9jey97axdvHzeG21NHH4trTwthKZrVGZLSUSpuPQJiGAI+GAI8SWKydycLL4d7f2OTi1+S9xNzL0ePT6OLGzeEAo5U0qJw/aLEAo5JFa7JBabEAp5Y4qZ2QxLyKmsm3kL2xoMOehrRNb7RIbbOZgrGre68AUqwAqZqNN5aKJ5N/lMq+qsd8kMa4pcWzh7muhLMEV69juq2kbKqgUaOTR5uMMZWLLZSGAI5VAIdEAH+ovNDHuNCnxcy3qcaYx8K8msGplrx+wLahjbYdXrV6vbMvYK9DrZ8QrZ8tqJuFms+Sos6sw8ecy8RffsNVeMCvmb43aLltv7Q4Y7EZWK4QWa1gt6meZKUdr6GOAZVeA4xPAISyveLUwtivxtKTpNJ2jcqfvcltiMiwwcfAoMVxhL+Kx7xjdrqTe60tsaNQs6KaRKACrJ6UTZwkqpqTL5pkHY4AloSgsd2ptNXPvNOOncuxxsqFl8lmg8apt8FJcr9EbryGxLqlkrkrY7dRa7ZGZLQ5t6iXUZ6PPpgVpZeJCJFKAIGareTa0+KJod3H0deY2M+esM25usmYu8d2zsJOdcBVvrCLbqcAOaaHaKQAMaScWqKBXqCXMJ2RHpiLF5NmJZAdAHN2kta11dKu1M+DkcZLdb+Mcql3TppyRJdzQ5ZtNZNlIY+DF4+voCOQAAAAZ3RSTlMABAT+MEEJ/RH+/TP+Zlv+pUo6Ifz8+fco/fz6+evr39S9nJmOilQaF/7+/f38+smmoYp6b1T+/v7++vj189zU0tDJxsGzsrKSfv34+Pf27dDOysG9t6+n/vv6+vr59uzr1tG+tZ6Qg9Ym3QAABR5JREFUSMeNlVVUG1EQhpcuxEspXqS0SKEtxQp1d3d332STTRpIQhIISQgJhODu7lAoDoUCpe7u7u7+1puGpqnCPOyZvffbOXPm/PsP9JfQgyCC+tmTABTOcbxDz/heENS7/1F+9nhvkHePG0wNDLbGWwdXL+rbLWvpmZHXD8+gMfBjTh+aSe6Gnn7lwQIOTR0c8wfX3PWgv7avbdKwf/ZoBp1Gp/PvuvXW3vw5ib7emnTW4OR+3D4jB9vjNJ/7gNvfWWeH/TO/JyYrsiKCRjVEZA3UB+96kON+DxOQ/NLE8PE5iUYgIXjFnCOlxEQMaSGVxjg4gxOnEycGz8bptuNjVx08LscIgrzH3umcn+KKtiBIyvzOO2O99aAdR8cF19oZalnCtvREUw79tCd5sow1g1UKM6kXqUx4T8wsi3sTjJ3yzDmmhenLXLpo8u45eG5y4Vvbk6kkC4LLtJMowkSQxmk4ggVJEG+7c6QpHT8vvW9X7/o7+3ELmiJi2mEzZJiz8cT6TBlanBk70cB5GGIGC1gRDdZ00yADLW1FL6gqhtvNXNG5S9gdSrk4M1qu7JAsmYshzDS4peoMrU/gT7qQdqYGZaYhxZmVbGJAm/CS/HloWyhRUlknQ9KYcExTwS80d3VNOxUZJpITYyspl0LbhArhpZCD9cRWEQuhYkNGMHToQ/2Cs6swJlb39CsllxdXX6IUKh/H5jbnSsPKjgmoaFQ1f8wRLR0UnGE/RcDEjj2jXG1WVTwUs8+zxfcrVO+vSsuOpVKxCfYZiQ0/aPKuxQbQ8lIz+DClxC8u+snlcJ7Yr1z1JPqUH0V+GDXbOwAib931Y4Imaq0NTIXPXY+N5L18GJ37SVWu+hwXff8l72Ds9XuwYIBaXPq6Shm4l+Vl/5QiOlV+uTk6YR9PxKsI9xNJny31ygK1e+nIRC1N97EGkFPI+jCpiHe5PCEy7oWqWSwRrpOvhFzcbTWMbm3ZJAOn1rUKpYIt/lDhW/5RHHteeWFN60qo98YJuoq1nK3uW5AabyspC1BcIEpOhft+SZAShYoLSvnmSfnYADUERP5jJn2h5XtsgCRuhYQqAvwTwn33+YWEKUI72HX5AtfSAZDe8F2DtPPm77afhl0EkthzuCQU0BWApgQIH9+KB0JhopMM7bJrdTRoleM2JAVNMyPF+wdoaz+XJpGoVAQ7WXUkcV7gT3oUZyi/ISIJAVKhgNp+4b4veCFhYVJw4locdSjZCp9cPUhLF9EZ3KKzURepMEtCDPP3VcWFx4UIiZIklIpFNfHpdEafIF2aRmOcrUmjohbT2WUllbmRvgfbythbQO3222fpDJoufaQPncYYuqoGtUEsCJZL6/3PR5b4syeSjZMQG/T2maGANlXT2v8S4AULWaUkCxfLyW8iW4kdka+nEMjxpL2NCwsYNBp+Q61PF43zyDg9Bm9+3NNySn78jMZUUkumqE4Gp7JmFOdP1vc8PpRrzj9+wPinCy8K1PiJ4aYbnTYpCCbDkBSbzhu2QJ1Gd82t8jI8TH51+OzvXoWbnXUOBkNW+0mWFwGcGOUVpU81/n3TOHb5oMt2FgYGjzau0Nif0Ss7Q3XB33hjjQHjHA5E5aOyIQc8CBrLdQSs3j92VG+3nNEjbkbdbBr9zm04ruvw37vh0QKOdeGIkckc80fX3KH/h7PT4BOjgCty8VZ5ux1MoO5Cf5naca2LAsEgehI+drX8o/0Nu+W0m6K/I9gGPd/dfx/EN/wN62AhsBWuAAAAAElFTkSuQmCC
">
<img alt="FlashAttention" src="https://img.shields.io/badge/%E2%9A%A1%EF%B8%8E%20FlashAttention-eae0c8?style=flat">
<img alt="SDPA" src="https://img.shields.io/badge/SDPA-DE3412?style=flat&logo=pytorch&logoColor=white">
</div>

## ๊ฐœ์š”[[overview]]

[`VisionEncoderDecoderModel`]์€ ์‚ฌ์ „ํ›ˆ๋ จ๋œ Transformer ๊ธฐ๋ฐ˜ ๋น„์ „ ๋ชจ๋ธ์„ ์ธ์ฝ”๋”๋กœ (*์˜ˆ:* [ViT](vit), [BEiT](beit), [DeiT](deit), [Swin](swin)),
์‚ฌ์ „ํ›ˆ๋ จ๋œ ์–ธ์–ด ๋ชจ๋ธ์„ ๋””์ฝ”๋”๋กœ (*์˜ˆ:* [RoBERTa](roberta), [GPT2](gpt2), [BERT](bert), [DistilBERT](distilbert)) ์‚ฌ์šฉํ•˜์—ฌ ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ๋ชจ๋ธ์„ ์ดˆ๊ธฐํ™”ํ•˜๋Š” ๋ฐ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์‚ฌ์ „ํ›ˆ๋ จ๋œ ์ฒดํฌํฌ์ธํŠธ๋กœ ์ด๋ฏธ์ง€-ํ…์ŠคํŠธ ์‹œํ€€์Šค ๋ชจ๋ธ์„ ์ดˆ๊ธฐํ™”ํ•˜๋Š” ํšจ๊ณผ๋Š” (์˜ˆ๋ฅผ ๋“ค์–ด) Minghao Li, Tengchao Lv, Lei Cui, Yijuan Lu, Dinei Florencio, Cha Zhang,
Zhoujun Li, Furu Wei์˜ [TrOCR: Transformer-based Optical Character Recognition with Pre-trained Models](https://huggingface.co/papers/2109.10282)์—์„œ ์ž…์ฆ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.

์ด๋Ÿฌํ•œ [`VisionEncoderDecoderModel`]์ด ํ›ˆ๋ จ/๋ฏธ์„ธ ์กฐ์ •๋œ ํ›„์—๋Š” ๋‹ค๋ฅธ ๋ชจ๋ธ๋“ค๊ณผ ๋งˆ์ฐฌ๊ฐ€์ง€๋กœ ์ €์žฅํ•˜๊ฑฐ๋‚˜ ๊ฐ€์ ธ์˜ฌ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค (์ž์„ธํ•œ ๋‚ด์šฉ์€ ์•„๋ž˜ ์˜ˆ์ œ๋ฅผ ์ฐธ์กฐํ•˜์„ธ์š”).

์‘์šฉ ์˜ˆ์‹œ๋กœ๋Š” ์ด๋ฏธ์ง€ ์บก์…”๋‹์ด ์žˆ์Šต๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์„œ๋Š” ์ธ์ฝ”๋”๊ฐ€ ์ด๋ฏธ์ง€๋ฅผ ์ธ์ฝ”๋”ฉํ•˜๊ณ , ์ดํ›„ ์ž๊ธฐํšŒ๊ท€ ์–ธ์–ด ๋ชจ๋ธ์ด ์บก์…˜์„ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค. ๋‹ค๋ฅธ ์˜ˆ์‹œ๋กœ๋Š” ๊ด‘ํ•™ ๋ฌธ์ž ์ธ์‹(OCR)์ด ์žˆ์Šต๋‹ˆ๋‹ค. [`VisionEncoderDecoderModel`]์˜ ์ธ์Šคํ„ด์Šค์ธ [TrOCR](trocr)์„ ์ฐธ์กฐํ•˜์„ธ์š”.

## ๋ชจ๋ธ ์„ค์ •์œผ๋กœ `VisionEncoderDecoderModel` ๋žœ๋ค ์ดˆ๊ธฐํ™”ํ•˜๊ธฐ[[randomly-initializing-visionencoderdecodermodel-from-model-configurations]]

[`VisionEncoderDecoderModel`]์€ ์ธ์ฝ”๋”์™€ ๋””์ฝ”๋” ์„ค์ •์„ ์ด์šฉํ•ด ๋žœ๋คํ•˜๊ฒŒ ๊ฐ€์ค‘์น˜๋ฅผ ์ดˆ๊ธฐํ™”ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ๋‹ค์Œ ์˜ˆ์ œ์—์„œ๋Š” ์ธ์ฝ”๋”์— ๊ธฐ๋ณธ [`ViTModel`] ์„ค์ •์„,
๋””์ฝ”๋”์— ๊ธฐ๋ณธ [`BertForCausalLM`] ์„ค์ •์„ ์‚ฌ์šฉํ•˜์—ฌ ์ด๋ฅผ ์ˆ˜ํ–‰ํ•˜๋Š” ๋ฐฉ๋ฒ•์„ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค.

```python
>>> from transformers import BertConfig, ViTConfig, VisionEncoderDecoderConfig, VisionEncoderDecoderModel

>>> config_encoder = ViTConfig()
>>> config_decoder = BertConfig()

>>> config = VisionEncoderDecoderConfig.from_encoder_decoder_configs(config_encoder, config_decoder)
>>> model = VisionEncoderDecoderModel(config=config)
```

## ์‚ฌ์ „ํ›ˆ๋ จ๋œ ์ธ์ฝ”๋”์™€ ๋””์ฝ”๋”์—์„œ `VisionEncoderDecoderModel` ์ดˆ๊ธฐํ™”ํ•˜๊ธฐ[[initialising-visionencoderdecodermodel-from-a-pretrained-encoder-and-a-pretrained-decoder]]

[`VisionEncoderDecoderModel`]์€ ์‚ฌ์ „ํ›ˆ๋ จ๋œ ์ธ์ฝ”๋” ์ฒดํฌํฌ์ธํŠธ์™€ ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ๋””์ฝ”๋” ์ฒดํฌํฌ์ธํŠธ์—์„œ ์ดˆ๊ธฐํ™”ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์‚ฌ์ „ ํ›ˆ๋ จ๋œ Transformer ๊ธฐ๋ฐ˜ ๋น„์ „ ๋ชจ๋ธ(*์˜ˆ:* [Swin](swin))์€ ์ธ์ฝ”๋” ์—ญํ• ์„ ํ•  ์ˆ˜ ์žˆ์œผ๋ฉฐ, ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ์˜คํ†  ์ธ์ฝ”๋”ฉ ๋ชจ๋ธ(*์˜ˆ:* BERT), ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ์ธ๊ณผ์  ์–ธ์–ด ๋ชจ๋ธ(*์˜ˆ:* GPT2), ๊ทธ๋ฆฌ๊ณ  ์‹œํ€€์Šค-ํˆฌ-์‹œํ€€์Šค ๋ชจ๋ธ์˜ ์‚ฌ์ „ ํ›ˆ๋ จ๋œ ๋””์ฝ”๋” ๋ถ€๋ถ„(*์˜ˆ:* BART์˜ ๋””์ฝ”๋”)์€ ๋ชจ๋‘ ๋””์ฝ”๋”๋กœ ์‚ฌ์šฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
๋””์ฝ”๋”๋กœ ์„ ํƒํ•˜๋Š” ์•„ํ‚คํ…์ฒ˜์— ๋”ฐ๋ผ ๊ต์ฐจ ์–ดํ…์…˜(cross-attention) ๋ ˆ์ด์–ด๊ฐ€ ๋žœ๋คํ•˜๊ฒŒ ์ดˆ๊ธฐํ™”๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
์‚ฌ์ „ ํ›ˆ๋ จ๋œ ์ธ์ฝ”๋”์™€ ๋””์ฝ”๋” ์ฒดํฌํฌ์ธํŠธ์—์„œ [`VisionEncoderDecoderModel`]์„ ์ดˆ๊ธฐํ™”ํ•˜๋ ค๋ฉด [*Warm-starting-encoder-decoder blog post*](https://huggingface.co/blog/warm-starting-encoder-decoder)์—์„œ ๋ณด์—ฌ์ค€ ๋ฐ”์™€ ๊ฐ™์ด ๋‹ค์šด์ŠคํŠธ๋ฆผ ์ž‘์—…์—์„œ ๋ชจ๋ธ์„ ๋ฏธ์„ธ ์กฐ์ •ํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.
์ด๋ฅผ ์œ„ํ•ด `VisionEncoderDecoderModel` ํด๋ž˜์Šค๋Š” [`VisionEncoderDecoderModel.from_encoder_decoder_pretrained`] ๋ฉ”์†Œ๋“œ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

```python
>>> from transformers import VisionEncoderDecoderModel

>>> model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(
... "microsoft/swin-base-patch4-window7-224-in22k", "google-bert/bert-base-uncased"
... )
```

## ๊ธฐ์กด `VisionEncoderDecoderModel` ์ฒดํฌํฌ์ธํŠธ ๋ถˆ๋Ÿฌ์™€ ์ถ”๋ก ํ•˜๊ธฐ[[loading-an-existing-visionencoderdecodermodel-checkpoint-and-perform-inference]]

`VisionEncoderDecoderModel` ํด๋ž˜์Šค์˜ ๋ฏธ์„ธ ์กฐ์ •๋œ ์ฒดํฌํฌ์ธํŠธ๋ฅผ ๊ฐ€์ ธ์˜ค๊ธฐ ์œ„ํ•ด, [`VisionEncoderDecoderModel`]์€ Transformers์˜ ๋‹ค๋ฅธ ๋ชจ๋ธ ์•„ํ‚คํ…์ฒ˜์™€ ๋งˆ์ฐฌ๊ฐ€์ง€๋กœ `from_pretrained(...)` ๋ฉ”์†Œ๋“œ๋ฅผ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

์ถ”๋ก ์„ ์ˆ˜ํ–‰ํ•˜๋ ค๋ฉด [`generate`] ๋ฉ”์†Œ๋“œ๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค. ์ด ๋ฉ”์†Œ๋“œ๋Š” ์ž๊ธฐํšŒ๊ท€์ ์œผ๋กœ ํ…์ŠคํŠธ๋ฅผ ์ƒ์„ฑํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์ด ๋ฉ”์†Œ๋“œ๋Š” ๊ทธ๋ฆฌ๋””(greedy), ๋น” ์„œ์น˜(beam search), ๋‹คํ•ญ ์ƒ˜ํ”Œ๋ง(multinomial sampling) ๋“ฑ ๋‹ค์–‘ํ•œ ํ˜•ํƒœ์˜ ๋””์ฝ”๋”ฉ์„ ์ง€์›ํ•ฉ๋‹ˆ๋‹ค.

```python
>>> import requests
>>> from PIL import Image

>>> from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel

>>> # ๋ฏธ์„ธ ์กฐ์ •๋œ ์ด๋ฏธ์ง€ ์บก์…”๋‹ ๋ชจ๋ธ๊ณผ ํ•ด๋‹น ํ† ํฌ๋‚˜์ด์ € ๋ฐ ์ด๋ฏธ์ง€ ํ”„๋กœ์„ธ์„œ๋ฅผ ๊ฐ€์ ธ์˜ต๋‹ˆ๋‹ค
>>> model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
>>> tokenizer = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
>>> image_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")

>>> # ์ด๋ฏธ์ง€์— ๋Œ€ํ•ด ์ถ”๋ก ์„ ์ˆ˜ํ–‰ํ•ด๋ณด๊ฒ ์Šต๋‹ˆ๋‹ค
>>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"
>>> image = Image.open(requests.get(url, stream=True).raw)
>>> pixel_values = image_processor(image, return_tensors="pt").pixel_values

>>> # ์ž๊ธฐํšŒ๊ท€์ ์œผ๋กœ ์บก์…˜ ์ƒ์„ฑ (๊ธฐ๋ณธ์ ์œผ๋กœ ๊ทธ๋ฆฌ๋”” ๋””์ฝ”๋”ฉ ์‚ฌ์šฉ)
>>> generated_ids = model.generate(pixel_values)
>>> generated_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
>>> print(generated_text)
a cat laying on a blanket next to a cat laying on a bed
```

## PyTorch ์ฒดํฌํฌ์ธํŠธ๋ฅผ `TFVisionEncoderDecoderModel`๋กœ ๊ฐ€์ ธ์˜ค๊ธฐ[[loading-a-pytorch-checkpoint-into-tfvisionencoderdecodermodel-]]

[`TFVisionEncoderDecoderModel.from_pretrained`]๋Š” ํ˜„์žฌ PyTorch ์ฒดํฌํฌ์ธํŠธ์—์„œ ๋ชจ๋ธ์„ ์ดˆ๊ธฐํ™”ํ•˜๋Š” ๊ฒƒ์„ ์ง€์›ํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค. ์ด ๋ฉ”์†Œ๋“œ์— `from_pt=True`๋ฅผ ์ „๋‹ฌํ•˜๋ฉด ์˜ˆ์™ธ๊ฐ€ ๋ฐœ์ƒํ•ฉ๋‹ˆ๋‹ค. ํŠน์ • ๋น„์ „ ์ธ์ฝ”๋”-๋””์ฝ”๋” ๋ชจ๋ธ์— PyTorch ์ฒดํฌํฌ์ธํŠธ๋งŒ ์žˆ๋Š” ๊ฒฝ์šฐ, ํ•ด๊ฒฐ ๋ฐฉ๋ฒ•์€ ๋‹ค์Œ๊ณผ ๊ฐ™์Šต๋‹ˆ๋‹ค:

```python
>>> from transformers import VisionEncoderDecoderModel, TFVisionEncoderDecoderModel

>>> _model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")

>>> _model.encoder.save_pretrained("./encoder")
>>> _model.decoder.save_pretrained("./decoder")

>>> model = TFVisionEncoderDecoderModel.from_encoder_decoder_pretrained(
... "./encoder", "./decoder", encoder_from_pt=True, decoder_from_pt=True
... )
>>> # ์ด๊ฒƒ์€ ์ด ํŠน์ • ๋ชจ๋ธ์˜ ๋ช‡ ๊ฐ€์ง€ ํŠน์ • ์†์„ฑ์„ ๋ณต์‚ฌํ•˜๊ธฐ ์œ„ํ•œ ๊ฒƒ์ž…๋‹ˆ๋‹ค.
>>> model.config = _model.config
```

## ํ›ˆ๋ จ[[training]]

๋ชจ๋ธ์ด ์ƒ์„ฑ๋˜๋ฉด, (์ด๋ฏธ์ง€, ํ…์ŠคํŠธ) ์Œ์œผ๋กœ ๊ตฌ์„ฑ๋œ ๋ฐ์ดํ„ฐ์…‹์„ ์‚ฌ์šฉํ•˜์—ฌ BART, T5 ๋˜๋Š” ๋‹ค๋ฅธ ์ธ์ฝ”๋”-๋””์ฝ”๋” ๋ชจ๋ธ๊ณผ ์œ ์‚ฌํ•˜๊ฒŒ ๋ฏธ์„ธ ์กฐ์ •ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
๋ณด์‹œ๋‹ค์‹œํ”ผ, ์†์‹ค์„ ๊ณ„์‚ฐํ•˜๊ธฐ ์œ„ํ•ด ๋ชจ๋ธ์— ํ•„์š”ํ•œ ์ž…๋ ฅ์€ ๋‹จ 2๊ฐœ์ž…๋‹ˆ๋‹ค: `pixel_values` (์ด๋ฏธ์ง€)์™€ `labels` (์ธ์ฝ”๋”ฉ๋œ ๋Œ€์ƒ ์‹œํ€€์Šค์˜ `input_ids`)์ž…๋‹ˆ๋‹ค.

```python
>>> from transformers import ViTImageProcessor, BertTokenizer, VisionEncoderDecoderModel
>>> from datasets import load_dataset

>>> image_processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
>>> tokenizer = BertTokenizer.from_pretrained("google-bert/bert-base-uncased")
>>> model = VisionEncoderDecoderModel.from_encoder_decoder_pretrained(
... "google/vit-base-patch16-224-in21k", "google-bert/bert-base-uncased"
... )

>>> model.config.decoder_start_token_id = tokenizer.cls_token_id
>>> model.config.pad_token_id = tokenizer.pad_token_id

>>> dataset = load_dataset("huggingface/cats-image")
>>> image = dataset["test"]["image"][0]
>>> pixel_values = image_processor(image, return_tensors="pt").pixel_values

>>> labels = tokenizer(
... "an image of two cats chilling on a couch",
... return_tensors="pt",
... ).input_ids

>>> # forward ํ•จ์ˆ˜๊ฐ€ ์ž๋™์œผ๋กœ ์˜ฌ๋ฐ”๋ฅธ decoder_input_ids๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค
>>> loss = model(pixel_values=pixel_values, labels=labels).loss
```

์ด ๋ชจ๋ธ์€ [nielsr](https://github.com/nielsrogge)์ด ๊ธฐ์—ฌํ–ˆ์Šต๋‹ˆ๋‹ค. ์ด ๋ชจ๋ธ์˜ TensorFlow์™€ Flax ๋ฒ„์ „์€
[ydshieh](https://github.com/ydshieh)๊ฐ€ ๊ธฐ์—ฌํ–ˆ์Šต๋‹ˆ๋‹ค.

## VisionEncoderDecoderConfig[[transformers.VisionEncoderDecoderConfig]]

[[autodoc]] VisionEncoderDecoderConfig

<frameworkcontent>
<pt>

## VisionEncoderDecoderModel[[transformers.VisionEncoderDecoderModel]]

[[autodoc]] VisionEncoderDecoderModel
- forward
- from_encoder_decoder_pretrained

</pt>
<tf>

## TFVisionEncoderDecoderModel[[transformers.TFVisionEncoderDecoderModel]]

[[autodoc]] TFVisionEncoderDecoderModel
- call
- from_encoder_decoder_pretrained

</tf>
<jax>

## FlaxVisionEncoderDecoderModel[[transformers.FlaxVisionEncoderDecoderModel]]

[[autodoc]] FlaxVisionEncoderDecoderModel
- __call__
- from_encoder_decoder_pretrained

</jax>
</frameworkcontent>