Disable HPU migration (future add-on to HF diffusers) for OH diffusers (#1866)

Signed-off-by: Daniel Socek <daniel.socek@intel.com>

Author: dsocek

optimum/habana/diffusers/pipelines/pipeline_utils.py

@@ -344,6 +344,16 @@ def is_saveable_module(name, value):
                 create_pr=create_pr,
             )
 
+    def to(self, *args, **kwargs):
+        """
+        Intercept to() method and disable gpu-hpu migration before sending to diffusers
+        """
+        kwargs["hpu_migration"] = False
+        return super().to(
+            *args,
+            **kwargs,
+        )
+
     @classmethod
     def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.PathLike]], **kwargs):
         """

optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py

@@ -260,6 +260,16 @@ def run_unet(
 
         return latents
 
+    def to(self, *args, **kwargs):
+        """
+        Intercept to() method and disable gpu-hpu migration before sending to diffusers
+        """
+        kwargs["hpu_migration"] = False
+        return super().to(
+            *args,
+            **kwargs,
+        )
+
     @classmethod
     def _split_inputs_into_batches(
         cls,