This commit is contained in:
Arthur Zucker 2024-03-05 18:27:25 +09:00
parent 132852203a
commit 0d52f9f582

View File

@ -22,7 +22,6 @@ from typing import List
import numpy as np import numpy as np
from transformers import PreTrainedTokenizer, PreTrainedTokenizerBase, PreTrainedTokenizerFast from transformers import PreTrainedTokenizer, PreTrainedTokenizerBase, PreTrainedTokenizerFast
from transformers.models.udop import UdopTokenizer, UdopTokenizerFast
from transformers.testing_utils import ( from transformers.testing_utils import (
require_pytesseract, require_pytesseract,
require_sentencepiece, require_sentencepiece,
@ -35,11 +34,11 @@ from transformers.utils import FEATURE_EXTRACTOR_NAME, cached_property, is_pytes
if is_torch_available(): if is_torch_available():
import torch import torch
from transformers import UdopTokenizer, UdopTokenizerFast
if is_pytesseract_available(): if is_pytesseract_available():
from PIL import Image from PIL import Image
from transformers import LayoutLMv3ImageProcessor, UdopProcessor from transformers import LayoutLMv3ImageProcessor, UdopProcessor