torch
numpy
diffuser
