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 慢十倍,而且只能推理不能训练。
但你想想,以前这玩意儿连门都进不来,现在至少能坐你桌边喝杯咖啡了。