Stability-AI / stable-fast-3d

SF3D: Stable Fast 3D Mesh Reconstruction with UV-unwrapping and Illumination Disentanglement
https://stable-fast-3d.github.io
Other
1.2k stars 131 forks source link

CPU Support like TripoSR? #12

Open KillyTheNetTerminal opened 3 months ago

vork commented 3 months ago

Most SF3D can already run on CPUs, except for texture baking, for which we're currently working on a solution. The last time I tested the pipeline, it took around 30 seconds on a CPU - take the value with a grain of salt, as it was quite some time ago.

KillyTheNetTerminal commented 3 months ago

Oh cool, but how can I run this on CPU what about bake with AMD GPU? like RX580

vork commented 3 months ago

We have recently pushed an update which enables CPU support. I'm not sure if AMD is supported now because I don't have such a GPU available.

jammm commented 3 months ago

Oh cool, but how can I run this on CPU what about bake with AMD GPU? like RX580

AMD support could work with this new version if you use Linux and the latest ROCm. As vork mentioned above however, we aren't sure because it's not tested. Having said that, RX 580 is very likely unsupported by ROCm at this point, so I wouldn't expect it to work unless it's a navi2 or above.

jammm commented 2 months ago

@KillyTheNetTerminal AMD support seems to be available in this fork of SF3D now https://github.com/harishanand95/stable-fast-3d. But you need Linux and ROCm that supports RX580 for it to work.