mirror of
https://github.com/mudler/LocalAI.git
synced 2026-06-25 00:59:28 -04:00
c0efc28968
Authors backend/python/trl/requirements-mps.txt and wires trl into the darwin CI matrix and gallery so the MPS training path can be built and validated on Apple Silicon. The MPS variant installs plain PyPI torch wheels (MPS-capable on macOS arm64) and the trl training stack; bitsandbytes is omitted as it is a CUDA-only dependency with poor Apple Silicon support. Signed-off-by: Ettore Di Giacinto <mudler@localai.io> Assisted-by: Claude:opus-4.8 [Claude Code]
13 lines
367 B
Plaintext
13 lines
367 B
Plaintext
torch==2.10.0
|
|
trl
|
|
peft
|
|
datasets>=3.0.0
|
|
transformers>=4.56.2
|
|
accelerate>=1.4.0
|
|
huggingface-hub>=1.3.0
|
|
sentencepiece
|
|
# Note: bitsandbytes is intentionally omitted on MPS. It is only used by the
|
|
# CUDA (cublas) variants for 8-bit/4-bit quantization and has poor support on
|
|
# Apple Silicon. torch here uses the plain PyPI wheels, which ship MPS support
|
|
# on macOS arm64.
|