accelerate
54370d45
- Adding `keep_torch_compile` argument to `unwrap_model` and `extract_model_from_parallel`. (#3282)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Adding `keep_torch_compile` argument to `unwrap_model` and `extract_model_from_parallel`. (#3282)
References
#3282 - Add `keep_torch_compile` param to `unwrap_model` and `extract_model_from_parallel` for distributed compiled model.
Author
ggoggam
Parents
d6d3e03c
Loading