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,
|