| torch==2.1.0 | |
| torchvision==0.16.0 | |
| diffusers==0.15.0 | |
| transformers==4.26.0 | |
| datasets==2.10.1 | |
| accelerate==0.22.0 | |
| gradio | |
| ftfy | |
| tensorboard | |
| Jinja2 | |
| tqdm | |
| scipy | |
| fire | |
| wandb | |
| safetensors | |
| opencv-python | |
| mediapipe | |
| pynvml | |
| cache | |
| xformers | |
| colorama | |
| torchmetrics | |
| lpips | |
| --extra-index-url https://download.pytorch.org/whl/cu118 | |