flax
replace bare asserts with ValueError in NNX attention validation
#5505

Merged

replace bare asserts with ValueError in NNX attention validation #5505

copybara-service merged 1 commit into google:main from mohsinm-dev:fix-attention-assert-to-valueerror

vfdev-5 commented on 2026-06-16

mohsinm-dev force pushed from a5984eaf to 407fb864 36 days ago

vfdev-5 commented on 2026-06-16

mohsinm-dev force pushed from 407fb864 to d2360a4e 36 days ago

replace bare asserts with ValueError in attention validation

4f85c88e

vfdev-5 force pushed from d2360a4e to 4f85c88e 34 days ago

vfdev-5 approved these changes on 2026-06-17

cgarciae added pull ready

cgarciae removed pull ready

cgarciae added pull ready

copybara-service merged 51b3d8f1 into main 25 days ago

Reviewers

vfdev-5

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone