Spaces:
Running
on
A10G
Running
on
A10G
Is there any way to run this space on CPU?
#3
by
4eJIoBek
- opened
Seems like just allowing only "-1" cuda device doesn't work
4eJIoBek
changed discussion title from
Is there any way to run this space on CPU
to Is there any way to run this space on CPU?
What about if I want to run it on my m1? Can I set torch.device("mps")?
Got it. line 22 in app.py
return MusicGen.get_pretrained(version, device='mps')
just if you did install https://developer.apple.com/metal/pytorch/ before
Ow, its really works for cpu too, thx
4eJIoBek
changed discussion status to
closed