accelerate
64580585 - FIX: Bug with unwrap_model and keep_fp32_wrapper=False (#1838)

Commit
2 years ago
FIX: Bug with unwrap_model and keep_fp32_wrapper=False (#1838) Using accelerator.unwrap_model(model, keep_fp32_wrapper=False) results in a defective forward method. This bug was (probably) introduced in PR #872. Wrapping the method in MethodType (as elsewhere in code) resolves the issue.
Parents
Loading