Skip to content

Commit 397cb63

Browse files
Niels RoggeNiels Rogge
authored andcommitted
Add model to appropriate place in toctree
1 parent 34c9f38 commit 397cb63

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

docs/source/en/_toctree.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,8 +218,6 @@
218218
title: DialoGPT
219219
- local: model_doc/distilbert
220220
title: DistilBERT
221-
- local: model_doc/donut
222-
title: Donut
223221
- local: model_doc/dpr
224222
title: DPR
225223
- local: model_doc/electra
@@ -429,6 +427,8 @@
429427
title: CLIP
430428
- local: model_doc/data2vec
431429
title: Data2Vec
430+
- local: model_doc/donut
431+
title: Donut
432432
- local: model_doc/flava
433433
title: FLAVA
434434
- local: model_doc/groupvit

src/transformers/models/donut/convert_donut_to_pytorch.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,8 +164,15 @@ def convert_donut_checkpoint(model_name, pytorch_dump_folder_path=None, push_to_
164164
task_prompt = task_prompt.replace("{user_input}", question)
165165
elif model_name == "naver-clova-ix/donut-base-finetuned-rvlcdip":
166166
task_prompt = "<s_rvlcdip>"
167+
elif model_name in [
168+
"naver-clova-ix/donut-base-finetuned-cord-v1",
169+
"naver-clova-ix/donut-base-finetuned-cord-v1-2560",
170+
]:
171+
task_prompt = "<s_cord>"
167172
elif model_name == "naver-clova-ix/donut-base-finetuned-cord-v2":
168173
task_prompt = "s_cord-v2>"
174+
elif model_name == "naver-clova-ix/donut-base-finetuned-zhtrainticket":
175+
task_prompt = "<s_zhtrainticket>"
169176
elif model_name in ["naver-clova-ix/donut-proto", "naver-clova-ix/donut-base"]:
170177
# use a random prompt
171178
task_prompt = "hello world"

0 commit comments

Comments
 (0)