torch
diffusers
from diffusers import AutoPipelineForText2Image
from diffusers import AutoPipelineForText2Image