网易首页 > 网易号 > 正文 申请入驻

在 Mac 上微调一切大模型

0
分享至

聊个让我非常兴奋的项目——mlx-tune

一句话概括:在你的 Mac 上,用 Unsloth 的 API 微调一切

LLM、视觉模型、TTS、STT、OCR、Embedding,全都能在 Apple Silicon 上本地微调

Mac 用户的微调困境

做大模型微调的同学应该都有过这种体验:想在本地跑个小实验验证下 idea,结果发现 Unsloth 依赖 Triton,而 Triton 不支持 Mac

于是你只剩两条路:

  1. 1. 花钱开云 GPU —— 就跑个 100 条数据的实验,有必要吗?

  2. 2. 用 mlx-lm 原生 API —— 但代码和 完全不兼容,到了云上还得重写一遍

mlx-tune(github.com/ARahim3/mlx-tune)的作者也遇到了一模一样的问题

他的解决思路非常简单粗暴:把 MLX 包装成 Unsloth 的 API

你在 Mac 上写的训练脚本,换个 import 就能直接在 CUDA 集群跑

# Unsloth (CUDA)                        # MLX-Tune (Apple Silicon)
from unsloth import FastLanguageModel from mlx_tune import FastLanguageModel
from trl import SFTTrainer from mlx_tune import SFTTrainer


# 后面的代码一模一样!

这才是真正解决问题的设计

下面这张图清楚展示了 mlx-tune 的工作流——本地原型验证,改个 import 就能上云训练:

功能有多全?看完吓一跳

它支持的训练方法比很多正经公司的内部工具都全:

语言模型训练:

  • • SFT :基础指令微调,这是最常用的

  • • DPO / ORPO / KTO / SimPO :各种偏好学习方法全覆盖

  • • GRPO :DeepSeek R1 风格的多生成 + 奖励训练

  • • CPT :持续预训练,支持解耦学习率

多模态训练:

  • • Vision :支持 Gemma 4、Qwen3.5、PaliGemma、LLaVA、Pixtral 等 VLM 微调

  • • TTS :Orpheus、OuteTTS、Spark-TTS、Sesame/CSM、Qwen3-TTS 五个 TTS 模型

  • • STT :Whisper、Moonshine、Qwen3-ASR、NVIDIA Canary、Voxtral 五个 STT 模型

  • • Embedding :BERT、ModernBERT、Qwen3-Embedding、Harrier,支持对比学习

  • • OCR :DeepSeek-OCR、GLM-OCR、olmOCR、Qwen-VL,内置 CER/WER 指标

进阶能力:

  • • MoE 微调 :支持 39+ 种 MoE 架构,包括 Qwen3.5-35B、Mixtral、DeepSeek 系列

  • • Gemma 4 Audio :12 层 Conformer 音频塔,原生处理 16kHz 音频

  • • LFM2 :Liquid AI 的混合卷积+GQA 架构

说真的,一个社区项目做到这个程度,相当离谱

全景架构一览——从 API 到硬件的五层设计:

安装很简单,推荐用 uv:

# 标准安装
uv pip install mlx-tune


# 带音频支持
uv pip install 'mlx-tune[audio]'
brew install ffmpeg

来个最基础的 SFT 微调示例:

from mlx_tune import FastLanguageModel, SFTTrainer, SFTConfig
from datasets import load_dataset

# 加载模型(4bit 量化,省显存)
model, tokenizer = FastLanguageModel.from_pretrained(
model_name="mlx-community/Llama-3.2-1B-Instruct-4bit",
max_seq_length=2048,
load_in_4bit=True,
)

# 加 LoRA
model = FastLanguageModel.get_peft_model(
model,
r=16,
target_modules=["q_proj", "k_proj", "v_proj", "o_proj"],
lora_alpha=16,
)

# 加载数据集
dataset = load_dataset("yahma/alpaca-cleaned", split="train[:100]")

# 训练
trainer = SFTTrainer(
model=model,
train_dataset=dataset,
tokenizer=tokenizer,
args=SFTConfig(
output_dir="outputs",
per_device_train_batch_size=2,
learning_rate=2e-4,
max_steps=50,
),
)
trainer.train()

# 保存:三种格式随你选
model.save_pretrained("lora_model") # LoRA 适配器
model.save_pretrained_merged("merged", tokenizer) # 合并后的完整模型
model.save_pretrained_gguf("model", tokenizer) # GGUF 格式,直接给 Ollama 用

如果你用过 Unsloth,这代码看着是不是特别眼熟?对,就是同一套 API

视觉模型微调

VLM 微调也是同样简洁的体验:

from mlx_tune import FastVisionModel, UnslothVisionDataCollator, VLMSFTTrainer
from mlx_tune.vlm import VLMSFTConfig

model, processor = FastVisionModel.from_pretrained(
"mlx-community/Qwen3.5-0.8B-bf16",
)

model = FastVisionModel.get_peft_model(
model,
finetune_vision_layers=True, # 视觉层也微调
finetune_language_layers=True,
r=16, lora_alpha=16,
)

# 训练(数据集格式和 Unsloth 一致)
FastVisionModel.for_training(model)
trainer = VLMSFTTrainer(
model=model,
tokenizer=processor,
data_collator=UnslothVisionDataCollator(model, processor),
train_dataset=dataset,
args=VLMSFTConfig(max_steps=30, learning_rate=2e-4),
)
trainer.train()

Gemma 4、Qwen3.5、PaliGemma、LLaVA、Pixtral 都支持

你甚至可以用 Vision GRPO 来训练视觉推理能力

TTS 微调:在 Mac 上克隆声音

这个功能我觉得特别有意思——在 Mac 上本地微调 TTS 模型:

from mlx_tune import FastTTSModel, TTSSFTTrainer, TTSSFTConfig, TTSDataCollator
from datasets import load_dataset, Audio

# 自动检测模型类型、编码器和 token 格式
model, tokenizer = FastTTSModel.from_pretrained(
"mlx-community/orpheus-3b-0.1-ft-bf16"
)
model = FastTTSModel.get_peft_model(model, r=16, lora_alpha=16)

dataset = load_dataset("MrDragonFox/Elise", split="train[:100]")
dataset = dataset.cast_column("audio", Audio(sampling_rate=24000))

trainer = TTSSFTTrainer(
model=model, tokenizer=tokenizer,
data_collator=TTSDataCollator(model, tokenizer),
train_dataset=dataset,
args=TTSSFTConfig(output_dir="./tts_output", max_steps=60),
)
trainer.train()

Orpheus、OuteTTS、Spark-TTS、Sesame/CSM、Qwen3-TTS 都支持

想做声音克隆或者风格化 TTS,再也不用租 GPU 了

工作流全景

mlx-tune 的定位非常清晰:本地原型 → 云端量产。

本地 Mac (mlx-tune)              云端 GPU (Unsloth)
├── 快速实验 ├── 大规模训练
├── 小数据集验证 ├── 完整数据集
├── 秒级迭代 ├── 生产级优化
└── 同一套代码 ─────────────────── └── 同一套代码

训练完还能直接导出:

  • • HuggingFace 格式 :标准保存

  • • GGUF :直接丢给 Ollama / llama.cpp

  • • push_to_hub :一键推到 HuggingFace Hub

它适合谁?

我觉得 mlx-tune 最适合这几类人:

  1. 1. Mac 用户 + 微调需求 :你有 M1/M2/M3/M4/M5,想在本地跑微调实验,这是最佳选择

  2. 2. 混合工作流用户 :本地调试、云端训练,代码无缝迁移

  3. 3. 多模态探索者 :想同时玩 LLM、Vision、TTS、STT、OCR 微调的人

  4. 4. 学习者 :想理解微调原理,在本地快速迭代比去 Colab 排队强太多

局限性也得说清楚:

  • • 训练速度肯定比不上 A100 + Unsloth,这是物理定律决定的

  • • GGUF 导出对量化模型有限制,建议用非量化基座模型

  • • 内存受限于 Mac 的统一内存(不过 Mac Studio 最高 512GB,够用了)

如果你是 Mac 用户,又对微调大模型感兴趣,强烈建议试试

制作不易,如果这篇文章觉得对你有用,可否点个关注。给我个三连击:点赞、转发和在看。若可以再给我加个,谢谢你看我的文章,我们下篇再见!

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
第二个许家印?又一地产大佬暴雷,涉案超200亿,现已被立案调查

第二个许家印?又一地产大佬暴雷,涉案超200亿,现已被立案调查

史之韵
2026-10-14 01:39:35
想让企退人员闭嘴,你敢亮明你的身份吗?让人看看你是个什么货色

想让企退人员闭嘴,你敢亮明你的身份吗?让人看看你是个什么货色

苗苗情感说
2026-10-16 09:55:15
国安部发出反诈警告!陌生电话接通先闭嘴,三秒沉默能保命!

国安部发出反诈警告!陌生电话接通先闭嘴,三秒沉默能保命!

白米饭怎么吃
2026-10-15 19:11:06
社保迎来最严新规,不交少交最高处罚3倍!

社保迎来最严新规,不交少交最高处罚3倍!

城事堂
2026-10-15 09:59:30
笑喷!松岛辉空抱怨:张本美和对我太严厉 训练比赛我说太累太困她都会骂我

笑喷!松岛辉空抱怨:张本美和对我太严厉 训练比赛我说太累太困她都会骂我

818体育
2026-10-16 11:59:06
马斯克的跑车开上了天!他说想活命的人不要买

马斯克的跑车开上了天!他说想活命的人不要买

果壳
2026-10-16 13:36:12
王曼昱的94万奖金,实际到手后,狠狠打脸饭圈“养活国乒”的猖狂

王曼昱的94万奖金,实际到手后,狠狠打脸饭圈“养活国乒”的猖狂

老鼜尾声电影解说
2026-10-15 11:53:41
邮轮开始离谱了!日韩航线人均从6000变成2880,全家开始海上漂了

邮轮开始离谱了!日韩航线人均从6000变成2880,全家开始海上漂了

众生的世界观
2026-10-14 08:40:02
苏州一对情侣,谈了7年,女子提了18次分手,分手后在街头痛哭!

苏州一对情侣,谈了7年,女子提了18次分手,分手后在街头痛哭!

川渝视觉
2026-04-17 22:13:14
最后一舞!伊藤美诚告别国际赛场,曾是日本第一个乒乓球奥运冠军

最后一舞!伊藤美诚告别国际赛场,曾是日本第一个乒乓球奥运冠军

观锐器
2026-10-15 18:24:48
再就业!ESPN官宣:林书豪基德加盟 担任NBA分析师+解说嘉宾

再就业!ESPN官宣:林书豪基德加盟 担任NBA分析师+解说嘉宾

醉卧浮生
2026-10-16 08:47:31
樊振东恋爱真相大白!在德国的现状曝光,难怪首次谈及退役及心愿

樊振东恋爱真相大白!在德国的现状曝光,难怪首次谈及退役及心愿

趣味八卦
2026-10-03 11:11:17
金价下跌!今日金价(10月15日),黄金价格多少钱一克

金价下跌!今日金价(10月15日),黄金价格多少钱一克

三农老历
2026-10-16 01:24:46
3种豆子,已被列入肝病“黑名单”,再爱吃也不要碰了

3种豆子,已被列入肝病“黑名单”,再爱吃也不要碰了

路医生健康科普
2026-09-28 16:52:58
10亿铂金级肉签发行申购,发行收益高,股民中签吃大肉!

10亿铂金级肉签发行申购,发行收益高,股民中签吃大肉!

数据挖掘分析
2026-10-16 07:08:51
“速成车”,正在消耗海外市场对中国汽车的信任

“速成车”,正在消耗海外市场对中国汽车的信任

虎嗅APP
2026-10-15 01:27:40
卢宇光称俄军训练场疑似发现鼠疫!欧盟通过对俄最大规模制裁

卢宇光称俄军训练场疑似发现鼠疫!欧盟通过对俄最大规模制裁

项鹏飞
2026-10-14 17:47:07
宏远早报!弗格表忠心,新大外今日体检,卢卡继承奎因财产!

宏远早报!弗格表忠心,新大外今日体检,卢卡继承奎因财产!

体坛卡卡说
2026-10-16 12:32:11
长江的"哭泣"!巨型鱤鱼在长江泛滥,十年禁渔却养虎为患?

长江的"哭泣"!巨型鱤鱼在长江泛滥,十年禁渔却养虎为患?

遁走的两轮
2026-09-23 08:17:48
杨振宁辞世1年,翁帆首谈感情,假若人生重来,她会舍弃这个念头

杨振宁辞世1年,翁帆首谈感情,假若人生重来,她会舍弃这个念头

冰语历史
2026-10-15 15:15:26
2026-10-16 16:00:49
Ai学习的老章 incentive-icons
Ai学习的老章
Ai学习的老章
3546文章数 11211关注度
往期回顾 全部

科技要闻

杨天若院士:AI是中国弯道超车的机会

头条要闻

会"飞"的特斯拉来了:中国官网开放预订 意向金33.2万

头条要闻

会"飞"的特斯拉来了:中国官网开放预订 意向金33.2万

体育要闻

工资一分不要,曼联名将想把母队带回西甲

娱乐要闻

徐艺洋官宣生子,完全看不出生过宝宝

财经要闻

特斯拉“飞”起来了!意向金33.2万元

汽车要闻

1213公里不充电 宝马这台纯电车真能跑这么远?

态度原创

亲子
教育
艺术
旅游
本地

亲子要闻

昊泽三步法解决孩子厌学复学率95%

教育要闻

给老姐上一课

艺术要闻

让人心醉的风景,德米特里·西索耶夫油画!

旅游要闻

高平文旅亮相“晋城好风光 青春向太行”高校行

本地新闻

因新荣记出圈,这座浙东小城才是吃货天堂

无障碍浏览 进入关怀版