accelerate

696e6cfe - Fix KeyError in extract_model_from_parallel for partial torch.compile (#3881)

Fix KeyError in extract_model_from_parallel for partial torch.compile (#3881) * Fix KeyError in extract_model_from_parallel for partial torch.compile * Add test for partial torch.compile unwrap in extract_model_from_parallel