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