pjlab-songcomposer / songcomposer

Apache License 2.0
162 stars 10 forks source link

What is the minimum requirement for GPU memory ? #4

Closed KevinGoodman closed 8 months ago

KevinGoodman commented 8 months ago

Thanks for sharing your excellent work. ~While attempting to execute inference.ipynb on an RTX3090 with 24GB, I encountered an Out of Memory error right at the onset, specifically when loading the model in the first cell. Is there a possibility to successfully run SongComposer on single/multiple commercial-grade GPUs like RTX3090/4090 ?~

just change .cuda().half() to .half().cuda() ...