j commited on
Commit
503abeb
1 Parent(s): 7b539b5

seed and waveform num parameters needed to be cast to int

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -38,7 +38,6 @@ pipe = AudioLDM2Pipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
38
  pipe = pipe.to("cuda")
39
 
40
 
41
-
42
  def process_fn(input_audio_path, prompt, negative_prompt, seed, num_inference_steps, audio_length_in_s, num_waveforms_per_prompt):
43
  """
44
  This function defines the audio processing steps
@@ -65,9 +64,9 @@ def process_fn(input_audio_path, prompt, negative_prompt, seed, num_inference_st
65
  audio = pipe(
66
  prompt,
67
  negative_prompt=negative_prompt,
68
- num_inference_steps=num_inference_steps,
69
  audio_length_in_s=audio_length_in_s,
70
- num_waveforms_per_prompt=num_waveforms_per_prompt,
71
  generator=generator,
72
  ).audios
73
 
 
38
  pipe = pipe.to("cuda")
39
 
40
 
 
41
  def process_fn(input_audio_path, prompt, negative_prompt, seed, num_inference_steps, audio_length_in_s, num_waveforms_per_prompt):
42
  """
43
  This function defines the audio processing steps
 
64
  audio = pipe(
65
  prompt,
66
  negative_prompt=negative_prompt,
67
+ num_inference_steps=int(num_inference_steps),
68
  audio_length_in_s=audio_length_in_s,
69
+ num_waveforms_per_prompt=int(num_waveforms_per_prompt),
70
  generator=generator,
71
  ).audios
72