diff --git a/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_backbone.py b/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_backbone.py
index 1abdbd5062..03ebb79757 100644
--- a/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_backbone.py
+++ b/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_backbone.py
@@ -550,6 +550,12 @@ def denoise_step(
         guidance_scale=None,
     ):
         step = ops.convert_to_tensor(step)
+        if not keras.utils.is_keras_tensor(num_steps):
+            num_steps = ops.convert_to_tensor(num_steps)
+        if guidance_scale is not None and not keras.utils.is_keras_tensor(
+            guidance_scale
+        ):
+            guidance_scale = ops.convert_to_tensor(guidance_scale)
         next_step = ops.add(step, 1)
         sigma, timestep = self.scheduler(step, num_steps)
         next_sigma, _ = self.scheduler(next_step, num_steps)