open-mmlab / FoleyCrafter

FoleyCrafter: Bring Silent Videos to Life with Lifelike and Synchronized Sounds. AI拟音大师,给你的无声视频添加生动而且同步的音效 😝
https://foleycrafter.github.io/
Apache License 2.0
282 stars 16 forks source link

Multi gpu? #4

Open matbee-eth opened 2 weeks ago

matbee-eth commented 2 weeks ago

I can't load (app.py) on my 24gb vram gpu, is there a way to split it across multiple cuda devices?

JonathanFly commented 2 weeks ago

Just skimming the code, seems like it may be possible to move just some of the models in CPU, and keep the slowest ones on GPU that fit within 24GB.