Roughly how much memory is needed to run this model

bytedance / ImageDream

The code releasing for https://image-dream.github.io/

Apache License 2.0

684 stars 30 forks source link

Open kingback156 opened 3 months ago

kingback156 commented 3 months ago

Roughly how much memory is needed to run this model? I find that he consumes a lot of memory each time causing an error to be reported.

denred0 commented 3 months ago

I got OUT OF MEMORY on A100 with 40Gb GPU.

denred0 commented 3 months ago

Sometimes I get OOM sometimes no. In general the results are pretty good and you can use treestudio mesh-exporter for a mesh extraction.