ed environment; your setup is automatically detected so you don’t need to explicitly define the rank or world_size. Move the DiffusionPipeline to distributed_state.device to assign a GPU to each process. Now use the split_between_processes utility as a context manager to automatically distribute the prompts between the number of processes.   Copied import torch
from accelerate import PartialState
from diffusers import DiffusionPipeline

pipelin