Skip to content

Add support for On-The-Fly Dynamic SafeTensors loading.#4286

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_938829595
Open

Add support for On-The-Fly Dynamic SafeTensors loading.#4286
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_938829595

Conversation

@copybara-service

@copybara-service copybara-service Bot commented Jun 26, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor

Add support for On-The-Fly Dynamic SafeTensors loading.

@codecov

codecov Bot commented Jun 26, 2026 •
edited
Loading

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 13.81579% with 262 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
...axtext/checkpoint_conversion/utils/load_dynamic.py 15.33% 138 Missing ⚠️
...ext/checkpoint_conversion/utils/tensor_handling.py 9.27% 88 Missing ⚠️
src/maxtext/checkpoint_conversion/utils/utils.py 13.88% 31 Missing ⚠️
src/maxtext/common/checkpointing.py 16.66% 5 Missing ⚠️

📢 Thoughts on this report? Let us know!

@copybara-service copybara-service Bot force-pushed the test_938829595 branch 3 times, most recently from 28871fa to bb21d1d Compare June 29, 2026 20:32
@abhinavclemson abhinavclemson changed the title Revised Add dynamic Hugging Face SafeTensors loading to MaxText Add support for On-The-Fly Dynamic SafeTensors loading. Jun 29, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@parambole parambole Awaiting requested review from parambole parambole is a code owner
@shuningjin shuningjin Awaiting requested review from shuningjin shuningjin is a code owner
@RissyRan RissyRan Awaiting requested review from RissyRan RissyRan is a code owner
@suexu1025 suexu1025 Awaiting requested review from suexu1025 suexu1025 is a code owner
@jiangjy1982 jiangjy1982 Awaiting requested review from jiangjy1982 jiangjy1982 is a code owner
@gobbleturk gobbleturk Awaiting requested review from gobbleturk gobbleturk is a code owner
@bvandermoon bvandermoon Awaiting requested review from bvandermoon bvandermoon is a code owner
@hengtaoguo hengtaoguo Awaiting requested review from hengtaoguo hengtaoguo is a code owner
@gagika gagika Awaiting requested review from gagika gagika is a code owner
@shralex shralex Awaiting requested review from shralex shralex is a code owner
@richjames0 richjames0 Awaiting requested review from richjames0 richjames0 is a code owner
@NicoGrande NicoGrande Awaiting requested review from NicoGrande NicoGrande is a code owner
@abhinavclemson abhinavclemson Awaiting requested review from abhinavclemson abhinavclemson is a code owner
@khatwanimohit khatwanimohit Awaiting requested review from khatwanimohit khatwanimohit is a code owner
@vipannalla vipannalla Awaiting requested review from vipannalla vipannalla is a code owner
@SurbhiJainUSC SurbhiJainUSC Awaiting requested review from SurbhiJainUSC SurbhiJainUSC is a code owner
@A9isha A9isha Awaiting requested review from A9isha A9isha is a code owner
@aireenmei aireenmei Awaiting requested review from aireenmei aireenmei is a code owner
@NuojCheng NuojCheng Awaiting requested review from NuojCheng NuojCheng is a code owner
@dipannita08 dipannita08 Awaiting requested review from dipannita08 dipannita08 is a code owner
@igorts-git igorts-git Awaiting requested review from igorts-git igorts-git is a code owner
@darisoy darisoy Awaiting requested review from darisoy darisoy is a code owner

At least 2 approving reviews are required to merge this pull request.

Assignees

@abhinavclemson abhinavclemson

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@abhinavclemson