17 lines
314 B
Plaintext
17 lines
314 B
Plaintext
torch
|
|
numpy
|
|
transformers==4.52.3
|
|
accelerate
|
|
qwen-omni-utils[decord]
|
|
librosa
|
|
Pillow
|
|
tqdm
|
|
scikit-learn
|
|
|
|
# Used by train_omni_turn_lora.py; already included in the ms-swift image.
|
|
peft
|
|
|
|
# Optional on compatible NVIDIA GPUs: flash-attn
|
|
# Optional CPU video backend: torchcodec
|
|
# Optional NVDEC backend: PyNvVideoCodec
|