transformers
diffusers==0.2.4
torch
spacy
ftfy
nvidia-ml-py3