refactor: change denoising_step_list type from List[int] to List[float] in InferenceConfig for improved flexibility

This commit is contained in:
gaclove
2025-11-26 12:00:17 +00:00
parent 9ad671b1c3
commit e57e93a91d
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -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:
+1 -1
View File
@@ -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