Update audiosr/pipeline.py
Browse files- audiosr/pipeline.py +0 -3
audiosr/pipeline.py
CHANGED
|
@@ -5,7 +5,6 @@ import yaml
|
|
| 5 |
import torch
|
| 6 |
import torchaudio
|
| 7 |
import numpy as np
|
| 8 |
-
import spaces
|
| 9 |
|
| 10 |
import audiosr.latent_diffusion.modules.phoneme_encoder.text as text
|
| 11 |
from audiosr.latent_diffusion.models.ddpm import LatentDiffusion
|
|
@@ -150,8 +149,6 @@ def build_model(ckpt_path=None, config=None, device=None, model_name="basic"):
|
|
| 150 |
|
| 151 |
return latent_diffusion
|
| 152 |
|
| 153 |
-
@spaces.GPU
|
| 154 |
-
|
| 155 |
def super_resolution(
|
| 156 |
latent_diffusion,
|
| 157 |
input_file,
|
|
|
|
| 5 |
import torch
|
| 6 |
import torchaudio
|
| 7 |
import numpy as np
|
|
|
|
| 8 |
|
| 9 |
import audiosr.latent_diffusion.modules.phoneme_encoder.text as text
|
| 10 |
from audiosr.latent_diffusion.models.ddpm import LatentDiffusion
|
|
|
|
| 149 |
|
| 150 |
return latent_diffusion
|
| 151 |
|
|
|
|
|
|
|
| 152 |
def super_resolution(
|
| 153 |
latent_diffusion,
|
| 154 |
input_file,
|