Modify bfloat16 handling in safetensors_loader.py to properly convert Torch bfloat16 views into numpy/JAX arrays. - #1936
Draft
copybara-service[bot] wants to merge 1 commit into
Draft
Modify bfloat16 handling in safetensors_loader.py to properly convert Torch bfloat16 views into numpy/JAX arrays. #1936copybara-service[bot] wants to merge 1 commit into
bfloat16 handling in safetensors_loader.py to properly convert Torch bfloat16 views into numpy/JAX arrays. #1936copybara-service[bot] wants to merge 1 commit into
Conversation
copybara-service
Bot
requested review from
abheesht17,
hgao327,
jiangyangmu,
lc5211,
s-noghabi,
sizhit2,
tianshub and
wang2yn84
as code owners
August 20, 2026 03:39
|
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 24, 2026 23:49
c0dadc1 to
6ff15c8
Compare
copybara-service
Bot
changed the title
bfloat16 handling in safetensors_loader.py to properly convert Torch bfloat16 views into numpy/JAX arrays.
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 24, 2026 23:59
6ff15c8 to
d0c2e0b
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 25, 2026 00:10
d0c2e0b to
0ccd77e
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 25, 2026 00:12
0ccd77e to
e1da978
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 25, 2026 00:22
e1da978 to
454a5ba
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 25, 2026 00:54
454a5ba to
8f43c27
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 25, 2026 15:57
e71cb85 to
f0c13d1
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 25, 2026 18:17
f0c13d1 to
02f19f3
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 01:45
02f19f3 to
f3521ba
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 14:46
f3521ba to
b52b033
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 16:14
b52b033 to
5cd1e0c
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 17:39
5cd1e0c to
9c0fff5
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 20:05
9c0fff5 to
cd82aa8
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 20:42
cd82aa8 to
8ce4b00
Compare
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 21:36
8ce4b00 to
d9a2b60
Compare
yatlas20098
marked this pull request as draft
August 26, 2026 21:38
…vert Torch bfloat16 views into numpy/JAX arrays. This fixes type conversion errors encountered when loading model checkpoint files containing bfloat16 tensors. PiperOrigin-RevId: 966870103
copybara-service
Bot
force-pushed
the
test_966870103
branch
from
August 26, 2026 22:32
d9a2b60 to
90e1a33
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Modify
bfloat16handling insafetensors_loader.pyto properly convert Torch bfloat16 views into numpy/JAX arrays.This fixes type conversion errors encountered when loading model checkpoint files containing bfloat16 tensors.