torch>=1.4
diffusers>=0.3.0