|
# Core Torch-related dependencies |
|
torch>=1.13.1 |
|
torchvision==0.15.2 |
|
torchaudio==2.0.2 |
|
|
|
# Scientific stack |
|
numpy==1.24.4 |
|
scipy==1.10.1 |
|
scikit-image==0.21.0 |
|
opencv-python==4.7.0.72 |
|
pillow==9.4.0 |
|
|
|
# Visualization & progress |
|
matplotlib==3.7.4 |
|
tqdm==4.64.1 |
|
|
|
# Transformers & Diffusers stack (compatible versions) |
|
transformers==4.33.3 |
|
diffusers==0.20.2 |
|
huggingface_hub==0.15.1 |
|
accelerate==0.20.3 # β
This version works well with the above stack |
|
|
|
# Other dependencies |
|
config==0.5.1 |
|
einops==0.7.0 |
|
onnxruntime==1.16.2 |
|
gradio==3.50.2 |
|
|
|
# Extra ML/deep learning utilities |
|
basicsr |
|
av |
|
fvcore |
|
cloudpickle |
|
omegaconf |
|
pycocotools |
|
|