mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
15 lines
654 B
Python
15 lines
654 B
Python
|
from typing import Union
|
||
|
|
|
||
|
|
|
||
|
|
def validate_weights(weights: Union[float, list[float]]) -> None:
|
||
|
|
"""Validate that all control weights in the valid range"""
|
||
|
|
to_validate = weights if isinstance(weights, list) else [weights]
|
||
|
|
if any(i < -1 or i > 2 for i in to_validate):
|
||
|
|
raise ValueError("Control weights must be within -1 to 2 range")
|
||
|
|
|
||
|
|
|
||
|
|
def validate_begin_end_step(begin_step_percent: float, end_step_percent: float) -> None:
|
||
|
|
"""Validate that begin_step_percent is less than end_step_percent"""
|
||
|
|
if begin_step_percent >= end_step_percent:
|
||
|
|
raise ValueError("Begin step percent must be less than or equal to end step percent")
|