返回首页
24小时热榜
技术产品经理的角色……很难定义,但我们都曾与优秀的产品经理和不优秀的产品经理合作过——通常在与他们合作几周后,就能很快判断出他们在这个 spectrum 上的位置。然而,我发现很难描述它们之间的区别。
我将微软的TRELLIS.2(4B参数图像到3D模型)移植到了Apple Silicon上,通过PyTorch MPS运行。原版需要CUDA以及flash_attn、nvdiffrast和自定义稀疏卷积内核,而这些在Mac上都无法使用。
我用纯PyTorch的替代方案替换了CUDA特定的操作:包括一个聚集-散布的稀疏3D卷积、用于稀疏变换器的SDPA注意力机制,以及一个基于Python的网格提取,替代了CUDA的哈希映射操作。总的修改涉及大约几百行代码,分布在9个文件中。
在M4 Pro(24GB)上,从单张照片生成约40万个顶点网格,耗时大约3.5分钟。虽然速度不及H100(只需几秒),但它可以离线运行,无需依赖云服务。
<a href="https://github.com/shivampkumar/trellis-mac" rel="nofollow">https://github.com/shivampkumar/trellis-mac</a>