Spaces:
Sleeping
Sleeping
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 |
|