torch==2.6.0
torchvision
transformers
imageio[ffmpeg]
safetensors
einops
modelscope
ftfy
pandas
accelerate
peft

[all]
av
torchaudio
torchcodec
librosa
bitsandbytes
comfy-kitchen
torchao>=0.16
deepspeed
tensorboard
swanlab
wandb

[audio]
av
torchaudio
torchcodec
librosa

[infiniteyou]
insightface
facexlib

[logger]
tensorboard
swanlab
wandb

[nexusgen]
qwen_vl_utils
transformers==4.49.0

[npu]
torch==2.7.1+cpu
torch-npu==2.7.1
torchvision==0.22.1+cpu

[npu_aarch64]
torch==2.7.1
torch-npu==2.7.1
torchvision==0.22.1

[quant]
bitsandbytes
comfy-kitchen
torchao>=0.16

[ses]
pywt

[training]
deepspeed