accelerate
54370d45 - Adding `keep_torch_compile` argument to `unwrap_model` and `extract_model_from_parallel`. (#3282)

Commit
1 year ago
Adding `keep_torch_compile` argument to `unwrap_model` and `extract_model_from_parallel`. (#3282)
Author
Parents
Loading