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