Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ from stable_audio_tools.inference.generation import generate_diffusion_cond
|
|
| 33 |
model, model_config = get_pretrained_model("ford442/stable-audio-open-1.0")
|
| 34 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 35 |
print(f"Using device: {device}")
|
| 36 |
-
model.to(device,torch.
|
| 37 |
|
| 38 |
@spaces.GPU(duration=60)
|
| 39 |
def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
|
|
|
|
| 33 |
model, model_config = get_pretrained_model("ford442/stable-audio-open-1.0")
|
| 34 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 35 |
print(f"Using device: {device}")
|
| 36 |
+
model.to(device,torch.float32)
|
| 37 |
|
| 38 |
@spaces.GPU(duration=60)
|
| 39 |
def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
|