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
8a450c3d
Unverified
Commit
8a450c3d
authored
Dec 04, 2024
by
hlky
Committed by
GitHub
Dec 04, 2024
Browse files
Fix `pipeline_stable_audio` formating (#10114)
parent
9ff72433
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py
...diffusers/pipelines/stable_audio/pipeline_stable_audio.py
+2
-1
No files found.
src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py
View file @
8a450c3d
...
...
@@ -34,6 +34,7 @@ from ...utils.torch_utils import randn_tensor
from
..pipeline_utils
import
AudioPipelineOutput
,
DiffusionPipeline
from
.modeling_stable_audio
import
StableAudioProjectionModel
if
is_torch_xla_available
():
import
torch_xla.core.xla_model
as
xm
...
...
@@ -732,7 +733,7 @@ class StableAudioPipeline(DiffusionPipeline):
if
callback
is
not
None
and
i
%
callback_steps
==
0
:
step_idx
=
i
//
getattr
(
self
.
scheduler
,
"order"
,
1
)
callback
(
step_idx
,
t
,
latents
)
if
XLA_AVAILABLE
:
xm
.
mark_step
()
...
...
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