PR #45912 Fix fsdp2 for gemma models with shared kv states

Fix fsdp2 for gemma models with shared kv states #45912

Cyrilvallez merged 2 commits into main from fix-gemmas

fix

80915cd2

Cyrilvallez force pushed from fb0301eb to 80915cd2 78 days ago

accelerate checks Mapping, not dict

30bebd21

Cyrilvallez merged 0d476f2c into main 77 days ago

Cyrilvallez deleted the fix-gemmas branch 77 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone