You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
tensor_parallel has a few distinct features compared to deepspeed:
It works with any PyTorch model out of the box. From what I've seen, deepspeed Automatic Tensor Parallelism only works for a selection of Hugging Face models.
It works in an interactive environment in a Jupyter Notebook. Deepspeed doesn't.
It seems that in deepspeed there is an auto tensor parallel scheme to use.
The text was updated successfully, but these errors were encountered: