Sayoyo commited on
Commit
b7f8342
·
1 Parent(s): 071dfa9

[feat] zero gpu support

Browse files
Files changed (1) hide show
  1. pipeline_ace_step.py +2 -1
pipeline_ace_step.py CHANGED
@@ -2,7 +2,7 @@ import random
2
  import time
3
  import os
4
  import re
5
-
6
  import torch
7
  import torch.nn as nn
8
  from loguru import logger
@@ -955,6 +955,7 @@ class ACEStepPipeline:
955
  latents, _ = self.music_dcae.encode(input_audio, sr=sr)
956
  return latents
957
 
 
958
  def __call__(
959
  self,
960
  audio_duration: float = 60.0,
 
2
  import time
3
  import os
4
  import re
5
+ import spaces
6
  import torch
7
  import torch.nn as nn
8
  from loguru import logger
 
955
  latents, _ = self.music_dcae.encode(input_audio, sr=sr)
956
  return latents
957
 
958
+ @spaces.GPU
959
  def __call__(
960
  self,
961
  audio_duration: float = 60.0,