We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
pipeline_stable_audio
1 parent 9ff7243 commit 8a450c3Copy full SHA for 8a450c3
src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py
@@ -34,6 +34,7 @@
34
from ..pipeline_utils import AudioPipelineOutput, DiffusionPipeline
35
from .modeling_stable_audio import StableAudioProjectionModel
36
37
+
38
if is_torch_xla_available():
39
import torch_xla.core.xla_model as xm
40
@@ -732,7 +733,7 @@ def __call__(
732
733
if callback is not None and i % callback_steps == 0:
734
step_idx = i // getattr(self.scheduler, "order", 1)
735
callback(step_idx, t, latents)
-
736
737
if XLA_AVAILABLE:
738
xm.mark_step()
739
0 commit comments