⭐ 559 スター | Python | ソフトウェアエンジニアリング向け機械学習プロジェクトプラットフォーム
MicrosoftのTRELLIS.2をApple SiliconでPyTorch MPSを使って動かせるようにしました。元のバージョンはCUDA専用の5つのコンパイル済み拡張機能（flex_gemm、flash_attn、o_voxel、cumesh、nvdiffrast）に依存していて、Macには対応版がありません。ゼロから代替バックエンドを作成しました：純粋なPyTorchの疎3D畳み込み（flex_gemmの代替）、空間ハッシュを使ったPythonメッシュ抽出（o_voxelのCUDAハッシュマップ操作の代替）、疎Transformerの注意機構（flash_attnの代替）、GPU加速三線形補間（cumeshとnvdiffrastの代替）です。
マイクロソフトの TRELLIS.2（40 億パラメータの画像から 3D へのモデル）を Apple Silicon 上で PyTorch MPS を使ってネイティブ実行するようにポートしました。元のバージョンは CUDA、flash_attn、nvdiffrast、カスタム疎畳み込みカーネルが必要でしたが、これらは Mac では動作しません。CUDA 固有の操作を純粋な PyTorch の代替案に置き換えました。具体的には、gather-scatter 疎 3D 畳み込み、疎 Transformer 用の SDPA アテンション、CUDA ハッシュマップ操作を置き換える Python ベースのメッシュ抽出です。合計 9 ファイルで数百行の変更を加えました。
M5 Proマシンに24GBのメモリを搭載しているユーザーが、Ollamaを使ってQwen 3.6の大規模言語モデル（LLM）を実行したいと考えています。Q4量子化（4ビット量子化）を使うべきか迷っていますが、適切なQ3量子化ソリューションが見つかりません。ハードウェアで快適に動作しながら、モデルの品質を過度に損なわないバージョンについて、アドバイスを求めています。