From 373b2c7c17881ce9707c6d6c4ae3e0b083466916 Mon Sep 17 00:00:00 2001 From: Daniel Socek Date: Wed, 10 Sep 2025 13:52:13 +0000 Subject: [PATCH] Disable SDP on BF16 default for generic diffusers HPU support Signed-off-by: Daniel Socek --- optimum/habana/diffusers/pipelines/pipeline_utils.py | 1 + .../stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py | 1 + 2 files changed, 2 insertions(+) diff --git a/optimum/habana/diffusers/pipelines/pipeline_utils.py b/optimum/habana/diffusers/pipelines/pipeline_utils.py index 3f1e5e22db..932f1a8b3f 100644 --- a/optimum/habana/diffusers/pipelines/pipeline_utils.py +++ b/optimum/habana/diffusers/pipelines/pipeline_utils.py @@ -349,6 +349,7 @@ def to(self, *args, **kwargs): Intercept to() method and disable gpu-hpu migration before sending to diffusers """ kwargs["hpu_migration"] = False + kwargs["sdp_on_bf16"] = False return super().to( *args, **kwargs, diff --git a/optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py b/optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py index edebec1778..8d82595846 100644 --- a/optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py +++ b/optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py @@ -286,6 +286,7 @@ def to(self, *args, **kwargs): Intercept to() method and disable gpu-hpu migration before sending to diffusers """ kwargs["hpu_migration"] = False + kwargs["sdp_on_bf16"] = False return super().to( *args, **kwargs,