diff --git a/optimum/habana/diffusers/models/controlnet_sdv.py b/optimum/habana/diffusers/models/controlnet_sdv.py
index 2cc62d7abb..fb7d77437b 100644
--- a/optimum/habana/diffusers/models/controlnet_sdv.py
+++ b/optimum/habana/diffusers/models/controlnet_sdv.py
@@ -325,7 +325,7 @@ def fn_recursive_add_processors(
             processors: Dict[str, AttentionProcessor],
         ):
             if hasattr(module, "get_processor"):
-                processors[f"{name}.processor"] = module.get_processor(return_deprecated_lora=True)
+                processors[f"{name}.processor"] = module.get_processor()
 
             for sub_name, child in module.named_children():
                 fn_recursive_add_processors(f"{name}.{sub_name}", child, processors)
diff --git a/optimum/habana/diffusers/schedulers/scheduling_euler_discrete.py b/optimum/habana/diffusers/schedulers/scheduling_euler_discrete.py
index 977b196e29..df87ed14a7 100644
--- a/optimum/habana/diffusers/schedulers/scheduling_euler_discrete.py
+++ b/optimum/habana/diffusers/schedulers/scheduling_euler_discrete.py
@@ -78,6 +78,8 @@ def __init__(
         prediction_type: str = "epsilon",
         interpolation_type: str = "linear",
         use_karras_sigmas: Optional[bool] = False,
+        use_exponential_sigmas: Optional[bool] = False,
+        use_beta_sigmas: Optional[bool] = False,
         sigma_min: Optional[float] = None,
         sigma_max: Optional[float] = None,
         timestep_spacing: str = "linspace",
@@ -94,6 +96,8 @@ def __init__(
             prediction_type,
             interpolation_type,
             use_karras_sigmas,
+            use_exponential_sigmas,
+            use_beta_sigmas,
             sigma_min,
             sigma_max,
             timestep_spacing,
diff --git a/setup.py b/setup.py
index 4249e21924..8985d49e7f 100644
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,7 @@
     "optimum",
     "torch",
     "accelerate >= 0.33.0, < 0.34.0",
-    "diffusers == 0.29.2",
+    "diffusers == 0.31.0",
     "huggingface_hub >= 0.24.7",
     "sentence-transformers == 3.2.1",
 ]