Ahmadzei's picture
update 1
57bdca5
raw
history blame contribute delete
253 Bytes
You don't need to explicitly place your model on a device.
from accelerate import Accelerator
accelerator = Accelerator()
Prepare to accelerate
The next step is to pass all the relevant training objects to the [~accelerate.Accelerator.prepare] method.