l/stable-diffusion-v1-5", torch_dtype=torch.float16, use_safetensors=True
) You’ll want to create a function to run inference; init_process_group handles creating a distributed environment with the type of backend to use, the rank of the current process, and the world_size or the number of processes participating. If you’re running inference in parallel over 2 GPUs, then the world_size is 2. Move the DiffusionPipeline to rank and use get_rank t