Spaces:
Running
on
Zero
Running
on
Zero
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):
|