1inkusFace commited on
Commit
f84fac1
·
verified ·
1 Parent(s): cc99eb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.bfloat16)
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):