Spaces:
Running
Running
Gong Junmin
commited on
Commit
·
a9fd5eb
1
Parent(s):
b0e2210
support audio2audio
Browse files- pipeline_ace_step.py +2 -2
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
|
|
@@ -989,7 +989,7 @@ class ACEStepPipeline:
|
|
| 989 |
latents, _ = self.music_dcae.encode(input_audio, sr=sr)
|
| 990 |
return latents
|
| 991 |
|
| 992 |
-
|
| 993 |
def __call__(
|
| 994 |
self,
|
| 995 |
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
|
|
|
|
| 989 |
latents, _ = self.music_dcae.encode(input_audio, sr=sr)
|
| 990 |
return latents
|
| 991 |
|
| 992 |
+
@spaces.GPU
|
| 993 |
def __call__(
|
| 994 |
self,
|
| 995 |
audio_duration: float = 60.0,
|