Skip to content

Commit 4662c4f

Browse files
authored
Dissable tatr loading warning (#365)
There should be no warning about some weights not being loaded when table-transformer is used Unstructured-IO/unstructured#3288
1 parent 939fd39 commit 4662c4f

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

CHANGELOG.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
## 0.7.37-dev0
1+
## 0.7.37-dev1
22

33
* refactor: remove layout analysis related code
4+
* enhancement: Hide warning about table transformer weights not being loaded
45

56
## 0.7.36
67

unstructured_inference/__version__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "0.7.37-dev0" # pragma: no cover
1+
__version__ = "0.7.37-dev1" # pragma: no cover

unstructured_inference/models/tables.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import numpy as np
1010
import torch
1111
from PIL import Image as PILImage
12-
from transformers import DetrImageProcessor, TableTransformerForObjectDetection
12+
from transformers import DetrImageProcessor, TableTransformerForObjectDetection, logging
1313
from transformers.models.table_transformer.modeling_table_transformer import (
1414
TableTransformerObjectDetectionOutput,
1515
)
@@ -65,7 +65,10 @@ def initialize(
6565

6666
try:
6767
logger.info("Loading the table structure model ...")
68+
cached_current_verbosity = logging.get_verbosity()
69+
logging.set_verbosity_error()
6870
self.model = TableTransformerForObjectDetection.from_pretrained(model)
71+
logging.set_verbosity(cached_current_verbosity)
6972
self.model.eval()
7073

7174
except EnvironmentError:

0 commit comments

Comments
 (0)