torch
transformers
diffusers
tensorflow
accelerate