BoxOfColors commited on
Commit
da46dda
·
1 Parent(s): 00e0606

Fix MMAudio TypeError: cast num_steps to int in _mmaudio_gpu_infer

Browse files

FlowMatching(num_steps=num_steps) was receiving a string from Gradio,
causing 'can only concatenate str (not int) to str' in torch.linspace.
Add int(num_steps) cast alongside the existing num_samples/crossfade_s casts.

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1054,6 +1054,7 @@ def _mmaudio_gpu_infer(video_file, prompt, negative_prompt, seed_val,
1054
 
1055
  seed_val = _resolve_seed(seed_val)
1056
  num_samples = int(num_samples)
 
1057
  crossfade_s = float(crossfade_s)
1058
 
1059
  device, dtype = _get_device_and_dtype()
 
1054
 
1055
  seed_val = _resolve_seed(seed_val)
1056
  num_samples = int(num_samples)
1057
+ num_steps = int(num_steps)
1058
  crossfade_s = float(crossfade_s)
1059
 
1060
  device, dtype = _get_device_and_dtype()