Handle model parallelism

by sgugger - opened Jun 22, 2023

←

Jun 22, 2023

With this added line (similar to many models in Transformers), this model will work with device_map="auto" during training.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Ready to merge

This branch is ready to get merged automatically.

· Sign up or log in to comment