Show HN: TRELLIS.2 image-to-3D가 Mac Silicon에서 실행 – Nvidia GPU 필요 없음
발행: (2026년 4월 20일 AM 09:07 GMT+9)
2 분 소요
원문: Hacker News
Source: Hacker News
Overview
Microsoft의 TRELLIS.2(4 B‑파라미터 이미지‑투‑3D 모델)를 Apple Silicon에서 PyTorch MPS를 이용해 포팅했습니다. 원래 구현은 CUDA와 flash_attn, nvdiffrast, 그리고 맞춤형 희소 컨볼루션 커널을 필요로 했으며, 이들은 macOS에서 작동하지 않았습니다.
Implementation Details
- CUDA‑전용 연산을 순수 PyTorch 대안으로 교체했습니다:
- 희소 3D 컨볼루션을 위한 gather‑scatter 연산.
- 희소 트랜스포머를 위한 SDPA 어텐션.
- CUDA 해시맵 연산을 대체하는 파이썬 기반 메쉬 추출.
- 변경 사항은 9개 파일에 걸쳐 수백 줄에 불과합니다.
Performance
- M4 Pro(24 GB)에서 단일 사진으로 약 400 K‑정점 메쉬를 3.5 분 정도에 생성합니다.
- H100 GPU가 몇 초 만에 처리하는 것보다 느리지만, 클라우드 의존 없이 오프라인으로 실행됩니다.