diff --git a/config_builder.py b/config_builder.py
index 97d557c..2352102 100644
--- a/config_builder.py
+++ b/config_builder.py
@@ -140,7 +140,7 @@ class InferenceConfigBuilder:
             steps_str = optional_params["denoising_steps"]
             if steps_str and steps_str.strip():
                 try:
-                    steps_list = [int(s.strip()) for s in steps_str.split(",")]
+                    steps_list = [float(s.strip()) for s in steps_str.split(",")]
                     config.denoising_step_list = steps_list
                     config.infer_steps = len(steps_list)
                 except ValueError:
diff --git a/data_models.py b/data_models.py
index 072b85b..2022f8b 100644
--- a/data_models.py
+++ b/data_models.py
@@ -53,7 +53,7 @@ class InferenceConfig:
     use_31_block: bool = True
 
     # Optional parameters
-    denoising_step_list: Optional[List[int]] = None
+    denoising_step_list: Optional[List[float]] = None
     resize_mode: str = "adaptive"
     fixed_area: str = "720p"
     segment_length: int = 81