accelerate
Fix KeyError in extract_model_from_parallel for partial torch.compile
#3881

Merged

Fix KeyError in extract_model_from_parallel for partial torch.compile #3881

SunMarc merged 2 commits into huggingface:main from amanzoni1:fix-extract-model-compiled-submodules

Fix KeyError in extract_model_from_parallel for partial torch.compile

f88344d8

Add test for partial torch.compile unwrap in extract_model_from_parallel

02ee93b4

SunMarc approved these changes on 2025-12-11

SunMarc merged 696e6cfe into main 27 days ago

Reviewers

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone