mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-01-14 16:47:55 -05:00
15 lines
665 B
Python
15 lines
665 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 or equal to 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")
|