AI Pulse

TRELLIS.2 现在能在 Mac 上本地跑,不用 NVIDIA 显卡了

TRELLIS.2 现在能在 Mac 上本地跑,不用 NVIDIA 显卡了

以前 TRELLIS.2 只认 NVIDIA 显卡,跟 Apple Silicon 的 Mac 完全绝缘。现在有人把整个模型搬到了 M 芯片上——不用外接显卡,也不用开虚拟机,直接本地跑。

移植不是简单打个补丁,而是把 CUDA 那套全拆了重做。哈希网格改用 Python 字典存,FlashAttention 换成 PyTorch 自带的 SDPA,连稀疏卷积都自己撸了个 gather-scatter 版本。这相当于把厨房从川菜馆改成粤菜馆,锅铲都得换。

M4 Pro 上跑一张图,三分钟出个八十多万个三角面的模型,OBJ 和 GLB 都有,PBR 材质骨架也搭好了,Blender 打开就能用。

可惜纹理还烤不了,因为 nvdiffrast 没人给写 MPS 版。网格偶尔漏个小洞,cumesh 功能还没跟上。速度嘛,比 NVIDIA 慢十倍,而且只能推理不能训练。

但你想想,以前这玩意儿连门都进不来,现在至少能坐你桌边喝杯咖啡了。

阅读原文
📚 相关主题 开源工程

📬 订阅 AI Pulse

每天三次更新,不错过重要信号

▲ 回到顶部