accelerate

03deec2a - Fix model copy after `dispatch_model` (#1971)

Fix model copy after `dispatch_model` (#1971) * Fix model copy after dispatch_model * Minor hook update to fix failing test * address reviewer comments