Skip to content

Commit 6afc995

Browse files
committed
Allow LayoutLMV3Processor to accept rescale_factor
1 parent 6f5a272 commit 6afc995

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

src/transformers/models/layoutlmv3/image_processing_layoutlmv3.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -188,10 +188,7 @@ def __init__(
188188
self.size = size
189189
self.resample = resample
190190
self.do_rescale = do_rescale
191-
if "rescale_factor" in kwargs:
192-
self.rescale_factor = kwargs["rescale_factor"]
193-
else:
194-
self.rescale_factor = rescale_value
191+
self.rescale_factor = kwargs.get("rescale_factor", rescale_value)
195192
self.do_normalize = do_normalize
196193
self.image_mean = image_mean if image_mean is not None else IMAGENET_STANDARD_MEAN
197194
self.image_std = image_std if image_std is not None else IMAGENET_STANDARD_STD

0 commit comments

Comments
 (0)