From 8a450c3da0b6a8aca1c36a4f3ea7f0096033cf56 Mon Sep 17 00:00:00 2001 From: hlky Date: Wed, 4 Dec 2024 12:17:42 +0000 Subject: [PATCH] Fix `pipeline_stable_audio` formating (#10114) --- src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py b/src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py index a30af53f77..cef63cf7e6 100644 --- a/src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py +++ b/src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py @@ -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()