pjlab-songcomposer / songcomposer

103 stars 3 forks source link

What is the minimum requirement for GPU memory ? #4

Closed KevinGoodman closed 3 months ago

KevinGoodman commented 3 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() ...