Skip to content

fix pissa test, convert to a workable in different hardware method - #10752

Draft
kahlun wants to merge 1 commit into
hiyouga:mainfrom
kahlun:pytest_pissa_fix
Draft

fix pissa test, convert to a workable in different hardware method#10752
kahlun wants to merge 1 commit into
hiyouga:mainfrom
kahlun:pytest_pissa_fix

Conversation

@kahlun

@kahlun kahlun commented Aug 13, 2026

Copy link
Copy Markdown
Contributor

What does this PR do?

Fixes # (issue)

Before submitting

…t for gauge-invariant comparison

Enhance compare_model function to support skipping specified keys and update test_pissa_train to reflect changes

Add assertions to compare_model and compare_model_pissa for key overlaps and weight matching
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant