'accelerate' missing in the model card.

#76
by Ina-Developer - opened

Running the model as suggested on the model card here on hugging face produced an error for me when I did so on a linux computer: 'ValueError: 'Using a 'device_map', 'tp.plan' 'torch.device' context manager or setting 'torch.set_default_device(device) requires 'accelerate''.

So instead of 'pip install -U transformers kernels torch'
one would need to run 'pip install -U transformers kernels torch accelerate' to run the model with the defaults in the code on the model card.

It's not a difficult issue to solve, but I still wonder why this package is missing in the model card.

Sign up or log in to comment