ximinng / SVGDreamer

[CVPR 2024] Official implementation for "SVGDreamer: Text Guided SVG Generation with Diffusion Model" https://arxiv.org/abs/2312.16476
https://ximinng.github.io/SVGDreamer-project/
MIT License
188 stars 14 forks source link

approximate gpu usage #6

Closed CodeGeek04 closed 5 months ago

CodeGeek04 commented 6 months ago

I tried running the code locally on my machine. I tried all different models available (for sd), but turns out my VRAM is not enought to test these. Can the authors maybe give few approximations for the resources required, or even share their system specifications used to train/inference the model.

ximinng commented 5 months ago

Hello CodeGeek04,

To run the examples in the README, you need a Nvidia V100 GPU with 30GB of VRAM. But if you're not satisfied, you can set vpsd.n_particle=1, vpsd.vsd_n_particle=1 and vpsd.phi_n_particle=1