Open nivibilla opened 6 months ago
Yes, you can. gpt-fast uses torchrun
to automatically dispatch tasks to different GPU nodes. You can use it just like regular python script.
Is notebook compatible with torchrun? Last time I heard it has limited (or none) multi-processing support.
Right, I think you have to find a notebook which support multiprocessing.
Hi,
Im trying to get an example working with Ray on Databricks. Essentially having multiple replicas of the model. Is it possible to load a model with tensor parallelism inside a notebook?
Thanks