2026-08-16 00:02:58 +08:00
|
|
|
torch
|
2026-09-10 06:54:22 +08:00
|
|
|
torchaudio
|
2026-08-16 00:02:58 +08:00
|
|
|
numpy
|
2026-08-27 15:24:59 +08:00
|
|
|
# Default for Thinker replay. The B1 local extractor also supports 4.57.6;
|
|
|
|
|
# keep that version in existing GPU containers (see docs/local_feature_baseline.md).
|
2026-08-16 00:02:58 +08:00
|
|
|
transformers==4.52.3
|
|
|
|
|
accelerate
|
|
|
|
|
qwen-omni-utils[decord]
|
|
|
|
|
librosa
|
|
|
|
|
Pillow
|
2026-08-16 22:19:43 +08:00
|
|
|
tqdm
|
|
|
|
|
scikit-learn
|
2026-09-10 06:54:22 +08:00
|
|
|
PyYAML
|
|
|
|
|
einops
|
2026-08-16 00:02:58 +08:00
|
|
|
|
2026-08-23 14:40:03 +08:00
|
|
|
# Used by train_omni_turn_lora.py; already included in the ms-swift image.
|
|
|
|
|
peft
|
|
|
|
|
|
2026-08-16 00:02:58 +08:00
|
|
|
# Optional on compatible NVIDIA GPUs: flash-attn
|
2026-08-23 16:41:05 +08:00
|
|
|
# Optional CPU video backend: torchcodec
|
2026-08-18 08:41:54 +08:00
|
|
|
# Optional NVDEC backend: PyNvVideoCodec
|