Commit
•
bb41997
1
Parent(s):
72c6a0c
Update modules/audio_ldm.py
Browse files- modules/audio_ldm.py +1 -0
modules/audio_ldm.py
CHANGED
|
@@ -11,6 +11,7 @@ class audio_generation:
|
|
| 11 |
|
| 12 |
def infer(self, prompt, num_length=10, negative="low quality, background sound, average quality, horrible quality"):
|
| 13 |
prompt = str(prompt)
|
|
|
|
| 14 |
if isinstance(num_length, str):
|
| 15 |
num_length = int(num_length)
|
| 16 |
audio = self.pipe(prompt, num_inference_steps=20, audio_length_in_s=num_length, negative_prompt=negative).audios[0]
|
|
|
|
| 11 |
|
| 12 |
def infer(self, prompt, num_length=10, negative="low quality, background sound, average quality, horrible quality"):
|
| 13 |
prompt = str(prompt)
|
| 14 |
+
num_length = str(num_length)
|
| 15 |
if isinstance(num_length, str):
|
| 16 |
num_length = int(num_length)
|
| 17 |
audio = self.pipe(prompt, num_inference_steps=20, audio_length_in_s=num_length, negative_prompt=negative).audios[0]
|