diffusers==0.31.0
transformers
peft
opencv-python
protobuf
sentencepiece
gradio
jupyter
torchao

lightning
datasets
torchvision
prodigyopt
wandb