torch
diffusers
omegaconf
