Skip to content

[Export] Add an option to not save the weights#5964

Merged
ManfeiBai merged 13 commits intomasterfrom
ManfeiBai-patch-43
Dec 4, 2023
Merged

[Export] Add an option to not save the weights#5964
ManfeiBai merged 13 commits intomasterfrom
ManfeiBai-patch-43

Conversation

@ManfeiBai
Copy link
Copy Markdown
Collaborator

@ManfeiBai ManfeiBai commented Dec 1, 2023

add an option to not save the weights, and check whether the weights exist before load

@ManfeiBai ManfeiBai changed the title [Export] Add an option to not save the weights [test][Export] Add an option to not save the weights Dec 1, 2023
@ManfeiBai ManfeiBai changed the title [test][Export] Add an option to not save the weights [Export] Add an option to not save the weights Dec 1, 2023
@ManfeiBai ManfeiBai requested a review from qihqi December 1, 2023 17:25
@ManfeiBai ManfeiBai marked this pull request as ready for review December 1, 2023 17:26
Comment thread torch_xla/stablehlo.py Outdated
Comment thread torch_xla/stablehlo.py Outdated
Comment thread test/stablehlo/test_stablehlo_save_load.py Outdated
Copy link
Copy Markdown
Collaborator

@qihqi qihqi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

One more nit, overall good to go. Thanks!

Comment thread torch_xla/stablehlo.py Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants