I had a similar issue and resolved it using pip install accelerate and reloading the notebook kernel I was using.
pip install accelerate