mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
53 lines
1.0 KiB
Plaintext
53 lines
1.0 KiB
Plaintext
# You need to use Python 3.11, the easiest way to get this on DGX OS without impacting the system version of Python is to create an environment with miniconda.
|
|
|
|
# specific dependency versions needed on DGX OS devices:
|
|
scipy==1.16.0
|
|
tifffile==2025.6.11
|
|
imageio==2.37.0
|
|
scikit_image==0.25.2
|
|
clean_fid==0.1.35
|
|
pywavelets==1.9.0
|
|
contourpy==1.3.3
|
|
opencv_python_headless==4.11.0.86
|
|
|
|
# additional dependencies matching requirements.txt:
|
|
torchao==0.10.0
|
|
safetensors
|
|
git+https://github.com/huggingface/diffusers@8600b4c10d67b0ce200f664204358747bd53c775
|
|
transformers==4.57.3
|
|
lycoris-lora==1.8.3
|
|
flatten_json
|
|
pyyaml
|
|
oyaml
|
|
tensorboard
|
|
kornia
|
|
invisible-watermark
|
|
einops
|
|
accelerate
|
|
toml
|
|
albumentations==1.4.15
|
|
albucore==0.0.16
|
|
pydantic
|
|
omegaconf
|
|
k-diffusion
|
|
open_clip_torch
|
|
timm
|
|
prodigyopt
|
|
controlnet_aux==0.0.10
|
|
python-dotenv
|
|
bitsandbytes
|
|
hf_transfer
|
|
lpips
|
|
pytorch_fid
|
|
optimum-quanto==0.2.4
|
|
sentencepiece
|
|
huggingface_hub
|
|
peft
|
|
gradio
|
|
python-slugify
|
|
opencv-python
|
|
pytorch-wavelets==1.3.0
|
|
matplotlib==3.10.1
|
|
setuptools==69.5.1
|
|
av==16.0.1
|
|
torchcodec |