Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
0a0fe69a
Unverified
Commit
0a0fe69a
authored
Aug 23, 2023
by
Sanchit Gandhi
Committed by
GitHub
Aug 23, 2023
Browse files
[AudioLDM Docs] Update docstring (#4744)
parent
124e76dd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
5 deletions
+4
-5
src/diffusers/pipelines/audioldm/pipeline_audioldm.py
src/diffusers/pipelines/audioldm/pipeline_audioldm.py
+4
-5
No files found.
src/diffusers/pipelines/audioldm/pipeline_audioldm.py
View file @
0a0fe69a
...
...
@@ -418,8 +418,7 @@ class AudioLDMPipeline(DiffusionPipeline):
Pre-generated negative text embeddings. Can be used to easily tweak text inputs (prompt weighting). If
not provided, `negative_prompt_embeds` are generated from the `negative_prompt` input argument.
return_dict (`bool`, *optional*, defaults to `True`):
Whether or not to return a [`~pipelines.stable_diffusion.StableDiffusionPipelineOutput`] instead of a
plain tuple.
Whether or not to return a [`~pipelines.AudioPipelineOutput`] instead of a plain tuple.
callback (`Callable`, *optional*):
A function that calls every `callback_steps` steps during inference. The function is called with the
following arguments: `callback(step: int, timestep: int, latents: torch.FloatTensor)`.
...
...
@@ -436,9 +435,9 @@ class AudioLDMPipeline(DiffusionPipeline):
Examples:
Returns:
[`~pipelines.
stable_diffusion.StableDiffusion
PipelineOutput`] or `tuple`:
If `return_dict` is `True`, [`~pipelines.
stable_diffusion.StableDiffus
io
n
PipelineOutput`] is returned,
otherwise a `tuple` is
returned where the first element is a list with the generated audio.
[`~pipelines.
Audio
PipelineOutput`] or `tuple`:
If `return_dict` is `True`, [`~pipelines.
Aud
ioPipelineOutput`] is returned,
otherwise a `tuple` is
returned where the first element is a list with the generated audio.
"""
# 0. Convert audio input length from seconds to spectrogram height
vocoder_upsample_factor
=
np
.
prod
(
self
.
vocoder
.
config
.
upsample_rates
)
/
self
.
vocoder
.
config
.
sampling_rate
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment