j
commited on
Commit
•
503abeb
1
Parent(s):
7b539b5
seed and waveform num parameters needed to be cast to int
Browse files
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 |
|