torch
diffusers
transformers
omegaconf
