| # PyTorch 2.4.1 + torchvision |
| torch==2.4.1 |
| torchvision==0.19.1 |
| torchaudio==2.4.1 |
|
|
| # Core dependencies |
| diffusers==0.29.0 |
| transformers==4.41.0 # ✅ This version supports huggingface-hub 1.x |
| accelerate==0.23.0 |
| pyrallis |
| easydict |
| omegaconf==2.1.1 |
| einops==0.3.0 |
| torch-fidelity==0.3.0 |
| torchmetrics==1.3.0 |
| pytorch-lightning==2.2.0 |
| taming-transformers @ git+https://github.com/CompVis/taming-transformers.git@master |
| clip @ git+https://github.com/OpenAI/CLIP.git@main |
|
|
| # Computer vision |
| opencv-python==4.5.5.64 |
| opencv-python-headless==4.5.5.64 |
| kornia==0.7.1 |
| imageio>=2.27 |
| imageio-ffmpeg==0.4.2 |
| matplotlib |
| scikit-image==0.21.0 |
| Pillow |
| segment-anything |
| numpy==1.24.4 |
|
|
| # Gradio |
| gradio |
| gradio-bbox-annotator |
|
|
| # Utilities |
| natsort |
| roifile |
| fill-voids |
| configargparse |
| ipywidgets |
| ftfy |
| sniffio |
| websocket-client |
| dask |
| tensorboard |
| joblib |
| lz4 |
| numba |
| edt |
| imagecodecs |
|
|
