refactor: change denoising_step_list type from List[int] to List[float] in InferenceConfig for improved flexibility
This commit is contained in:
+1
-1
@@ -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
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user