Spaces:

chenxie95
/

MeanAudio

Running on Zero

AndreasXi commited on 12 days ago

Commit

c5a0b30

1 Parent(s): ba762df

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ def generate_audio_gradio(
     seed,
     variant,
 ):
-    dtype = torch.float32
     if duration <= 0 or num_steps <= 0:
         raise ValueError("Duration and number of steps must be positive.")
     if variant not in all_model_cfg:

     seed,
     variant,
 ):
+    dtype = torch.bfloat16
     if duration <= 0 or num_steps <= 0:
         raise ValueError("Duration and number of steps must be positive.")
     if variant not in all_model_cfg: