Spaces:
Runtime error
Runtime error
[feat] zero gpu support
Browse files- 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,
|