triton
pydantic
accelerate
imageio
numpy
omegaconf
tokenizers
transformers
diffusers
sentencepiece