requirements.txt

accelerate
bitsandbytes
diffusers
doki_theme_jupyter
einops
gradio
huggingface_hub
numpy
opencv_python
packaging
Pillow
torch
torchinfo
torchvision
tqdm
transformers
git+https://github.com/wdcqc/WaveFunctionDiffusion.git