open-mmlab / FoleyCrafter

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

Multi gpu? #4

Open matbee-eth opened 4 months ago

matbee-eth commented 4 months 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 4 months 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.