Skip to content

P
pdf-miner
Commit 00ad8e67 authored by myhloli's avatar myhloli
Browse files
Options

fix(magic_pdf):disable torchtext deprecation warning >=0.18.0

parent 230e37ed
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
magic_pdf/model/pdf_extract_kit.py
View file @ 00ad8e67
...@@ -9,7 +9,9 @@ try: ...@@ -9,7 +9,9 @@ try:
import argparse import argparse
import numpy as np import numpy as np
import torch import torch
import torchtext;torchtext.disable_torchtext_deprecation_warning() import torchtext
if torchtext.__version__ >= "0.18.0":
torchtext.disable_torchtext_deprecation_warning()
from PIL import Image from PIL import Image
from torchvision import transforms from torchvision import transforms
from torch.utils.data import Dataset, DataLoader from torch.utils.data import Dataset, DataLoader
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment