ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Separate dependencies for DGX OS devices (#610)

Browse Source
This commit is contained in:
raziel2001au
2025-12-27 01:32:26 +10:00
committed by GitHub
parent a767b82b60
commit 90fc99f486
2 changed files with 50 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
dgx_requirements.txt
View File

@@ -10,5 +10,42 @@ pywavelets==1.9.0
contourpy==1.3.3
opencv_python_headless==4.11.0.86
# we include the base requirements.txt for the remaining dependencies:
-r requirements.txt
# additional dependencies matching requirements.txt:
torchao==0.10.0
safetensors
git+https://github.com/huggingface/diffusers@6bf668c4d217ebc96065e673d8a257fd79950d34
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