Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix UDOP dtype issue #34180

Merged
merged 11 commits into from
Oct 18, 2024
+2 −2
Merged

Fix UDOP dtype issue #34180

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
4441d40
Trigger UDOP tests
Rocketknight1 Oct 15, 2024
75aaa0a
Try forcing dtype in LayoutLMV3
Rocketknight1 Oct 15, 2024
0bb721e
Do checks to see where uint8 is getting in
Rocketknight1 Oct 15, 2024
242a680
Do checks to see where uint8 is getting in
Rocketknight1 Oct 15, 2024
3674df5
Found it!
Rocketknight1 Oct 15, 2024
f0fa8f8
Add .astype(np.float32)
Rocketknight1 Oct 15, 2024
d1f432f
Remove forced check, make fixup
Rocketknight1 Oct 15, 2024
4f855d3
Checking where exactly the uint8 creeps in
Rocketknight1 Oct 18, 2024
d7f0720
More checking on the uint8 issues
Rocketknight1 Oct 18, 2024
1c40330
Manually upcast in rescale()
Rocketknight1 Oct 18, 2024
8cfa5d0
Remove UDOP trigger
Rocketknight1 Oct 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/transformers/image_transforms.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -123,11 +123,11 @@ def rescale(
if not isinstance(image, np.ndarray):
raise TypeError(f"Input image must be of type np.ndarray, got {type(image)}")

rescaled_image = image * scale
rescaled_image = image.astype(np.float64) * scale # Numpy type promotion has changed, so always upcast first
if data_format is not None:
rescaled_image = to_channel_dimension_format(rescaled_image, data_format, input_data_format)

rescaled_image = rescaled_image.astype(dtype)
rescaled_image = rescaled_image.astype(dtype) # Finally downcast to the desired dtype at the end

return rescaled_image

Expand Down
Loading