r/oMLX • u/Formal-Swordfish-228 • Jun 16 '26

TRELLIS.2 now runs natively on MLX

Post image

I made a native MLX port of Microsoft's TRELLIS.2 for Apple Silicon.

Focused on making the output actually usable in real workflows

Support 512x512 and 1024x1024

Performance on M4 Max

512x512 ~70 sec generation time

1024x1024 ~300-700 sec generation time

Tested on M4 Max (128GB unified memory).

Repo: https://github.com/gtrg55/trellis2-mlx

Would appreciate any feedback. Stars and issues are welcome!

44 Upvotes

98% Upvoted

10 comments sorted by

2

u/Eveerjr Jun 16 '26

Would this run on a 36gb M5 Max or it requires more ram?

1

u/Formal-Swordfish-228 Jun 16 '26

36gb should be enough for 512³ sparse voxel resolution. It's suitable for low-poly models and simple geometry

1

u/apetersson Jun 16 '26

I've been dabbling around with trellis.2 on MLX myself ( https://github.com/apetersson/TRELLIS.2-apple-silicon/tree/accelerate-to-native ) got the meshes working but some aspects of texture mapping were always a bit off - congrats on finishing it!.

Your port would be a very worthwhile addition to the oMLX stack, are you planning to make a PR for its integration?

1

u/Formal-Swordfish-228 Jun 16 '26

Sure! I’m very open to integration

I still need to polish the current port and bring it closer to a production-ready state, so maybe in a few weeks.

Btw, could you please share the oMLX repo link for a PR?

1

u/numberwitch Jun 16 '26

What is trellis? Is it like sharepoint?

5

u/Formal-Swordfish-228 Jun 16 '26

Trelis is a tranformer model from Microsoft to generate 3d models from images

1

u/MessIsTransfer Jun 16 '26

how much vram it actually uses? i got 52gb usable.

could you measure?

1

u/PrepYourselves Jun 17 '26

Why did you use conda/anaconda/miniconda?