#pytorch==2.1.0
#pytorch-cuda==11.8
#torchaudio=2.0.2
#torchtriton=2.0.0
#torchvision=0.15.2

accelerate==0.25.0
torch==2.0.1
torchmetrics==1.2.1
tqdm==4.66.1
transformers==4.36.2
diffusers==0.25.0
einops==0.7.0
bitsandbytes==0.39.0
scipy==1.11.1
opencv-python
gradio==4.36.1
fvcore
cloudpickle
omegaconf
pycocotools
basicsr
av
onnxruntime==1.16.2