From e57e93a91d75913aacf4897a229dde0c7c52d99e Mon Sep 17 00:00:00 2001
From: gaclove <peng.gaoc@gmail.com>
Date: Thu, 20 Nov 2025 16:26:44 +0800
Subject: [PATCH] refactor: change denoising_step_list type from List[int] to
 List[float] in InferenceConfig for improved flexibility

---
 config_builder.py | 2 +-
 data_models.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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