Преглед на файлове

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

myhloli преди 1 година
родител
ревизия
00ad8e67ff
променени са 1 файла, в които са добавени 3 реда и са изтрити 1 реда
  1. 3 1
      magic_pdf/model/pdf_extract_kit.py

+ 3 - 1
magic_pdf/model/pdf_extract_kit.py

@@ -9,7 +9,9 @@ try:
     import argparse
     import numpy as np
     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 torchvision import transforms
     from torch.utils.data import Dataset, DataLoader